Update flacuna.py
Browse files- flacuna.py +1 -1
flacuna.py
CHANGED
|
@@ -31,7 +31,7 @@ if __name__ == "__main__":
|
|
| 31 |
)
|
| 32 |
model = get_peft_model(model, lora_config)
|
| 33 |
|
| 34 |
-
weight = torch.load("
|
| 35 |
model.load_state_dict(weight)
|
| 36 |
|
| 37 |
prompt = (
|
|
|
|
| 31 |
)
|
| 32 |
model = get_peft_model(model, lora_config)
|
| 33 |
|
| 34 |
+
weight = torch.load("pytorch_model.bin", map_location="cpu")
|
| 35 |
model.load_state_dict(weight)
|
| 36 |
|
| 37 |
prompt = (
|