jena-shreyas commited on
Commit
06ef027
·
1 Parent(s): 8427fe9

Fix issue where Qwen model_name was passed as model_path for loading

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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,