Spaces:

VirtualOasis
/

Academic-chat

Runtime error

c commited on Jan 22, 2024

Commit

02c6443

verified ·

1 Parent(s): 955ca24

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from peft import PeftModel, PeftConfig
 base_model = "mistralai/Mistral-7B-v0.1"
-# config = PeftConfig.from_pretrained("kiki7sun/mixtral-academic-finetune0119")
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1",
                                              low_cpu_mem_usage=True,
                                              return_dict=True,

 base_model = "mistralai/Mistral-7B-v0.1"
+config = PeftConfig.from_pretrained("kiki7sun/mixtral-academic-finetune0119")
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1",
                                              low_cpu_mem_usage=True,
                                              return_dict=True,