Update model_loader.py
Browse files- model_loader.py +1 -1
model_loader.py
CHANGED
|
@@ -26,7 +26,7 @@ def load_model_bundle():
|
|
| 26 |
# ===== CM2 model =====
|
| 27 |
ckpt_path = os.environ.get(
|
| 28 |
"CKPT_PATH",
|
| 29 |
-
"assets/checkpoints/
|
| 30 |
)
|
| 31 |
|
| 32 |
cfg = ModelCfg(cm_mode="CM2") # pastikan cfg ini IDENTIK training
|
|
|
|
| 26 |
# ===== CM2 model =====
|
| 27 |
ckpt_path = os.environ.get(
|
| 28 |
"CKPT_PATH",
|
| 29 |
+
"assets/checkpoints/split_90_05_05__CM2.pt"
|
| 30 |
)
|
| 31 |
|
| 32 |
cfg = ModelCfg(cm_mode="CM2") # pastikan cfg ini IDENTIK training
|