Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,6 +19,19 @@ from transformers import (
|
|
| 19 |
from qwen_vl_utils import process_vision_info
|
| 20 |
from taxonomy import policy_v1
|
| 21 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
# Set up logging
|
| 23 |
logging.basicConfig(
|
| 24 |
level=logging.INFO,
|
|
|
|
| 19 |
from qwen_vl_utils import process_vision_info
|
| 20 |
from taxonomy import policy_v1
|
| 21 |
|
| 22 |
+
import random
|
| 23 |
+
import numpy as np
|
| 24 |
+
|
| 25 |
+
SEED = 42 # choose any number you like
|
| 26 |
+
random.seed(SEED)
|
| 27 |
+
np.random.seed(SEED)
|
| 28 |
+
torch.manual_seed(SEED)
|
| 29 |
+
if torch.cuda.is_available():
|
| 30 |
+
torch.cuda.manual_seed_all(SEED)
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
|
| 34 |
+
|
| 35 |
# Set up logging
|
| 36 |
logging.basicConfig(
|
| 37 |
level=logging.INFO,
|