Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from peft import PeftModel
|
|
| 8 |
# CONFIG
|
| 9 |
# -------------------------
|
| 10 |
BASE_MODEL_ID = "HuggingFaceTB/SmolVLM2-256M-Video-Instruct"
|
| 11 |
-
FINETUNED_MODEL_ID = "Chaste20/smolvlm2-asl-ql"
|
| 12 |
|
| 13 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
DTYPE = torch.bfloat16 if torch.cuda.is_available() else torch.float32
|
|
|
|
| 8 |
# CONFIG
|
| 9 |
# -------------------------
|
| 10 |
BASE_MODEL_ID = "HuggingFaceTB/SmolVLM2-256M-Video-Instruct"
|
| 11 |
+
FINETUNED_MODEL_ID = "Chaste20/smolvlm2-asl-ql-2"
|
| 12 |
|
| 13 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
DTYPE = torch.bfloat16 if torch.cuda.is_available() else torch.float32
|