FaiziRBLX commited on
Commit
c7e4431
Β·
verified Β·
1 Parent(s): 5a8c847

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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('indonesian_llm_model (46).pt')}")
24
- if not os.path.exists('indonesian_llm_model (46).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 (46).pt') / 1e6:.1f} MB")
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("indonesian_llm_model (46).pt", map_location='cpu', weights_only=False)
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 (42).pt')}")
24
+ if not os.path.exists('indonesian_llm_model (42).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 (42).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 (42).pt", map_location='cpu', weights_only=False)
37
  logger.info(f"Checkpoint keys: {list(checkpoint.keys())}")
38
 
39
  logger.info("Building model...")