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
Files changed (1) hide show
  1. main.py +1 -1
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-quantized-new.pth")
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