Spaces:

Priyanka6
/

fine-tuning-inference

Runtime error

Priyanka6 commited on Feb 25

Commit

5277acd

1 Parent(s): eb72a06

Update space

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,7 +85,7 @@ def trim_adapter_weights(model_path):
     # if not os.path.exists(model_path):
     #     raise FileNotFoundError(f"Adapter file not found: {model_path}")
-    checkpoint = torch.load(model_path, map_location="cpu")
     key_to_trim = "lm_head.lora_B.default.weight"

     # if not os.path.exists(model_path):
     #     raise FileNotFoundError(f"Adapter file not found: {model_path}")
+    checkpoint = torch.load(model_path, map_location="cpu",weights_only=False)
     key_to_trim = "lm_head.lora_B.default.weight"