Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ class ModelManager:
|
|
50 |
def load_model(self, model_config):
|
51 |
if model_config['name'] not in self.models:
|
52 |
try:
|
53 |
-
self.models[model_config['name']] = Llama.from_pretrained(
|
54 |
except Exception as e:
|
55 |
print(f"Error loading model {model_config['name']}: {e}")
|
56 |
|
|
|
50 |
def load_model(self, model_config):
|
51 |
if model_config['name'] not in self.models:
|
52 |
try:
|
53 |
+
self.models[model_config['name']] = Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename'])
|
54 |
except Exception as e:
|
55 |
print(f"Error loading model {model_config['name']}: {e}")
|
56 |
|