AurelioAguirre commited on
Commit
eda2ff2
·
1 Parent(s): 1cdb3dd

checking checkpoint dir

Browse files
Files changed (1) hide show
  1. main/main.py +1 -0
main/main.py CHANGED
@@ -76,6 +76,7 @@ async def initialize_model(request: InitializeRequest):
76
  # Get the project root directory (where main.py is located)
77
  project_root = Path(__file__).parent
78
  checkpoints_dir = project_root / "checkpoints"
 
79
 
80
  # For LitGPT downloaded models, path includes organization
81
  if "/" in request.model_path:
 
76
  # Get the project root directory (where main.py is located)
77
  project_root = Path(__file__).parent
78
  checkpoints_dir = project_root / "checkpoints"
79
+ logger.info(f"Checkpoint dir is: {checkpoints_dir}")
80
 
81
  # For LitGPT downloaded models, path includes organization
82
  if "/" in request.model_path: