LukasHug commited on
Commit
6cd2959
·
verified ·
1 Parent(s): 3871a0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
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,