Spaces:
Sleeping
Sleeping
Commit
·
06ef027
1
Parent(s):
8427fe9
Fix issue where Qwen model_name was passed as model_path for loading
Browse files
app.py
CHANGED
|
@@ -56,8 +56,11 @@ def load_model_with_quantization(
|
|
| 56 |
print(f"Loading {model_name} with {quantization} quantization...")
|
| 57 |
model_path = model_name
|
| 58 |
# Load the HF version of LLaVA-Video-7B instead of the default version, for transformers v5 compatibility
|
|
|
|
| 59 |
if model_name == "LLaVA-Video-7B-Qwen2":
|
| 60 |
model_path = "Isotr0py/LLaVA-Video-7B-Qwen2-hf"
|
|
|
|
|
|
|
| 61 |
model = load_model(
|
| 62 |
model_path,
|
| 63 |
device_map=DEVICE_MAP,
|
|
|
|
| 56 |
print(f"Loading {model_name} with {quantization} quantization...")
|
| 57 |
model_path = model_name
|
| 58 |
# Load the HF version of LLaVA-Video-7B instead of the default version, for transformers v5 compatibility
|
| 59 |
+
# For the Qwen models, load the model from the Qwen directory
|
| 60 |
if model_name == "LLaVA-Video-7B-Qwen2":
|
| 61 |
model_path = "Isotr0py/LLaVA-Video-7B-Qwen2-hf"
|
| 62 |
+
elif model_name.startswith("Qwen"):
|
| 63 |
+
model_path = f"Qwen/{model_name}"
|
| 64 |
model = load_model(
|
| 65 |
model_path,
|
| 66 |
device_map=DEVICE_MAP,
|