robinhad commited on
Commit
99989a6
1 Parent(s): 7ff53cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model = LlamaForCausalLM.from_pretrained(
21
  "openlm-research/open_llama_3b_v2",
22
  **options
23
  )
24
- model = PeftModel.from_pretrained(model, "robinhad/open_llama_3b_uk", offload_folder="")
25
 
26
 
27
  def generate_prompt(instruction, input=None, output=""):
 
21
  "openlm-research/open_llama_3b_v2",
22
  **options
23
  )
24
+ model = PeftModel.from_pretrained(model, "robinhad/open_llama_3b_uk", **options)
25
 
26
 
27
  def generate_prompt(instruction, input=None, output=""):