Upload folder using huggingface_hub
Browse files
test.py
CHANGED
|
@@ -20,7 +20,6 @@ with safe_open(checkpoint_path, framework="pt", device="cpu") as f:
|
|
| 20 |
state_dict = {key: f.get_tensor(key) for key in f.keys()}
|
| 21 |
|
| 22 |
model.load_state_dict(state_dict)
|
| 23 |
-
print("Веса успешно загружены из safetensors")
|
| 24 |
|
| 25 |
model.eval()
|
| 26 |
|
|
|
|
| 20 |
state_dict = {key: f.get_tensor(key) for key in f.keys()}
|
| 21 |
|
| 22 |
model.load_state_dict(state_dict)
|
|
|
|
| 23 |
|
| 24 |
model.eval()
|
| 25 |
|