Spaces:
Running
Running
abhisheksan
commited on
Commit
•
9e5e405
1
Parent(s):
a96ed1b
Update model file path in ModelManager; change from poeticagpt-quantized-new.pth to poeticagpt.pth
Browse files
main.py
CHANGED
@@ -79,7 +79,7 @@ class ModelManager:
|
|
79 |
logger.warning(f"Could not load custom vocabulary: {str(e)}")
|
80 |
|
81 |
logger.info("Loading model...")
|
82 |
-
model_path = os.path.join(BASE_MODEL_DIR, "poeticagpt
|
83 |
if not os.path.exists(model_path):
|
84 |
logger.error(f"Model file not found at {model_path}")
|
85 |
return False
|
|
|
79 |
logger.warning(f"Could not load custom vocabulary: {str(e)}")
|
80 |
|
81 |
logger.info("Loading model...")
|
82 |
+
model_path = os.path.join(BASE_MODEL_DIR, "poeticagpt.pth")
|
83 |
if not os.path.exists(model_path):
|
84 |
logger.error(f"Model file not found at {model_path}")
|
85 |
return False
|