wormcode commited on
Commit
8479fa1
·
1 Parent(s): aa3df59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -65,6 +65,7 @@ def main(
65
  )
66
  model = PeftModel.from_pretrained(
67
  model,
 
68
  adapter_name=lora_weights,
69
  device_map={"": device},
70
  )
 
65
  )
66
  model = PeftModel.from_pretrained(
67
  model,
68
+ "./",
69
  adapter_name=lora_weights,
70
  device_map={"": device},
71
  )