Spaces:

Priyanka6
/

fine-tuning-inference

Runtime error

Priyanka6 commited on Feb 25

Commit

9e3a75e

1 Parent(s): b1c1bda

Update space

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,9 +87,7 @@ def trim_adapter_weights(model_path):
     #     raise FileNotFoundError(f"Adapter file not found: {model_path}")
     checkpoint = load_file(model_path)
-    print('beginning of checkpoint')
-    print(checkpoint)
-    print('checkpoint end')
     key_to_trim = "lm_head.lora_B.default.weight"

     #     raise FileNotFoundError(f"Adapter file not found: {model_path}")
     checkpoint = load_file(model_path)
+    print("Keys in checkpoint:", list(checkpoint.keys()))
     key_to_trim = "lm_head.lora_B.default.weight"