Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
| 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
|