Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ class ModelManager:
|
|
52 |
def __init__(self):
|
53 |
self.models = []
|
54 |
self.loaded = False
|
55 |
-
|
56 |
@spaces.GPU(duration=0)
|
57 |
def load_model(self, model_config):
|
58 |
return {"model": Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename']), "name": model_config['name']}
|
|
|
52 |
def __init__(self):
|
53 |
self.models = []
|
54 |
self.loaded = False
|
55 |
+
|
56 |
@spaces.GPU(duration=0)
|
57 |
def load_model(self, model_config):
|
58 |
return {"model": Llama.from_pretrained(repo_id=model_config['repo_id'], filename=model_config['filename']), "name": model_config['name']}
|