Update app.py
Browse files
app.py
CHANGED
|
@@ -20,10 +20,10 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
| 20 |
logger.info(f"Device: {device}")
|
| 21 |
|
| 22 |
# ββ Cek model file ββββββββββββββββββββββββββββββββββββββββ
|
| 23 |
-
logger.info(f"model.pt ada: {os.path.exists('
|
| 24 |
-
if not os.path.exists('
|
| 25 |
raise FileNotFoundError("model.pt tidak ditemukan! Upload dulu ke Space.")
|
| 26 |
-
logger.info(f"model.pt size: {os.path.getsize('
|
| 27 |
|
| 28 |
# ββ Load tokenizer ββββββββββββββββββββββββββββββββββββββββ
|
| 29 |
logger.info("Loading tokenizer...")
|
|
@@ -33,7 +33,7 @@ logger.info("Tokenizer OK")
|
|
| 33 |
|
| 34 |
# ββ Load model ββββββββββββββββββββββββββββββββββββββββββββ
|
| 35 |
logger.info("Loading checkpoint...")
|
| 36 |
-
checkpoint = torch.load("
|
| 37 |
logger.info(f"Checkpoint keys: {list(checkpoint.keys())}")
|
| 38 |
|
| 39 |
logger.info("Building model...")
|
|
|
|
| 20 |
logger.info(f"Device: {device}")
|
| 21 |
|
| 22 |
# ββ Cek model file ββββββββββββββββββββββββββββββββββββββββ
|
| 23 |
+
logger.info(f"model.pt ada: {os.path.exists('indonesian_llm_model (39).pt')}")
|
| 24 |
+
if not os.path.exists('indonesian_llm_model (39).pt'):
|
| 25 |
raise FileNotFoundError("model.pt tidak ditemukan! Upload dulu ke Space.")
|
| 26 |
+
logger.info(f"model.pt size: {os.path.getsize('indonesian_llm_model (39).pt') / 1e6:.1f} MB")
|
| 27 |
|
| 28 |
# ββ Load tokenizer ββββββββββββββββββββββββββββββββββββββββ
|
| 29 |
logger.info("Loading tokenizer...")
|
|
|
|
| 33 |
|
| 34 |
# ββ Load model ββββββββββββββββββββββββββββββββββββββββββββ
|
| 35 |
logger.info("Loading checkpoint...")
|
| 36 |
+
checkpoint = torch.load("indonesian_llm_model (39).pt", map_location='cpu', weights_only=False)
|
| 37 |
logger.info(f"Checkpoint keys: {list(checkpoint.keys())}")
|
| 38 |
|
| 39 |
logger.info("Building model...")
|