Spaces:
Paused
Paused
Commit
·
eda2ff2
1
Parent(s):
1cdb3dd
checking checkpoint dir
Browse files- 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:
|