Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,6 @@ def load_videollama3_model():
|
|
| 54 |
torch_dtype=torch.float16,
|
| 55 |
trust_remote_code=True,
|
| 56 |
low_cpu_mem_usage=True,
|
| 57 |
-
attn_implementation="flash_attention_2"
|
| 58 |
)
|
| 59 |
|
| 60 |
model_loaded = True
|
|
|
|
| 54 |
torch_dtype=torch.float16,
|
| 55 |
trust_remote_code=True,
|
| 56 |
low_cpu_mem_usage=True,
|
|
|
|
| 57 |
)
|
| 58 |
|
| 59 |
model_loaded = True
|