zhiqiulin commited on
Commit
7f0b452
·
verified ·
1 Parent(s): 67afa2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -6,6 +6,11 @@ from t2v_metrics import VQAScore, list_all_vqascore_models
6
  import torch
7
  torch.jit.script = lambda f: f
8
 
 
 
 
 
 
9
  print(list_all_vqascore_models())
10
 
11
  # Initialize the model only once
 
6
  import torch
7
  torch.jit.script = lambda f: f
8
 
9
+ import os
10
+ os.environ['PYTORCH_JIT'] = '0'
11
+ print(f"PYTORCH_JIT: {os.environ['PYTORCH_JIT']}")
12
+
13
+
14
  print(list_all_vqascore_models())
15
 
16
  # Initialize the model only once