SiyouLi commited on
Commit
f5bd3b0
·
verified ·
1 Parent(s): ec7f461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ DEFAULT_MODEL_ID = os.environ.get("QTSPLUS_MODEL_ID", "AlpachinoNLP/QTSplus-3B")
31
  DEFAULT_QUESTION = "What is happening in the video?"
32
 
33
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
34
- DTYPE = torch.bfloat16 if DEVICE.type == "cuda" else torch.float32
35
 
36
  _MODEL: Optional[AutoModelForCausalLM] = None
37
  _PROCESSOR: Optional[AutoProcessor] = None
 
31
  DEFAULT_QUESTION = "What is happening in the video?"
32
 
33
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
34
+ DTYPE = torch.bfloat16 if DEVICE.type == "cuda" else torch.float16
35
 
36
  _MODEL: Optional[AutoModelForCausalLM] = None
37
  _PROCESSOR: Optional[AutoProcessor] = None