Update app.py
Browse files
app.py
CHANGED
|
@@ -8,8 +8,7 @@ from best import ModelConfig, IndonesianLLM # <--- Ganti MyTransformer dengan na
|
|
| 8 |
|
| 9 |
# 1. Load Tokenizer (Sesuai dengan yang ada di best.py)
|
| 10 |
# 1. Load file bobotnya dulu ke dalam variabel
|
| 11 |
-
state_dict = torch.load("model.pt", map_location=torch.device('cpu'))
|
| 12 |
-
|
| 13 |
# 2. Buat kamus baru untuk menyimpan nama kunci yang sudah diperbaiki
|
| 14 |
new_state_dict = {}
|
| 15 |
for key, value in state_dict.items():
|
|
|
|
| 8 |
|
| 9 |
# 1. Load Tokenizer (Sesuai dengan yang ada di best.py)
|
| 10 |
# 1. Load file bobotnya dulu ke dalam variabel
|
| 11 |
+
state_dict = torch.load("model.pt", map_location=torch.device('cpu'), weights_only=False)
|
|
|
|
| 12 |
# 2. Buat kamus baru untuk menyimpan nama kunci yang sudah diperbaiki
|
| 13 |
new_state_dict = {}
|
| 14 |
for key, value in state_dict.items():
|