Spaces:

bstraehle
/

sft

Running

bstraehle commited on Aug 4

Commit

8c90204

•

1 Parent(s): 17cccab

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -149,7 +149,7 @@ def load_model(model_name):
         #task_type="CAUSAL_LM",
     )
     print("222")
-    model = PeftModel.from_pretrained(base_model, "new_model", peft_config=peft_config)
     print("333")
     model = model.merge_and_unload()
     print("444")

         #task_type="CAUSAL_LM",
     )
     print("222")
+    model = PeftModel.from_pretrained(model, "new_model", peft_config=peft_config)
     print("333")
     model = model.merge_and_unload()
     print("444")