DioulaD commited on
Commit
1afc5b2
1 Parent(s): b8172a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def load_peft_model():
12
  trust_remote_code=True,
13
  offload_folder="./offload"
14
  )
15
- model = PeftModel.from_pretrained(model, peft_model_id, offload_dir="")
16
  model = model.merge_and_unload()
17
 
18
  config = PeftConfig.from_pretrained(peft_model_id)
 
12
  trust_remote_code=True,
13
  offload_folder="./offload"
14
  )
15
+ model = PeftModel.from_pretrained(model, peft_model_id, offload_folder="./offload")
16
  model = model.merge_and_unload()
17
 
18
  config = PeftConfig.from_pretrained(peft_model_id)