Spaces:
Configuration error
Configuration error
Update agent.py
Browse files
agent.py
CHANGED
@@ -141,9 +141,6 @@ def initialize_models(use_api_mode=False):
|
|
141 |
top_p=0.95,
|
142 |
top_k=20
|
143 |
)
|
144 |
-
else:
|
145 |
-
# Use regular Qwen model if Pixtral not found
|
146 |
-
raise ImportError("Pixtral 12B not found")
|
147 |
except (ImportError, Exception) as e:
|
148 |
print(f"Error loading Pixtral with vLLM: {e}")
|
149 |
print(f"Falling back to {pixtral_model} with HuggingFace...")
|
|
|
141 |
top_p=0.95,
|
142 |
top_k=20
|
143 |
)
|
|
|
|
|
|
|
144 |
except (ImportError, Exception) as e:
|
145 |
print(f"Error loading Pixtral with vLLM: {e}")
|
146 |
print(f"Falling back to {pixtral_model} with HuggingFace...")
|