AkashDataScience commited on
Commit
d0c796a
·
1 Parent(s): 1d6ade2

Updated file path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model_kwargs = dict(
14
  )
15
  base_model = AutoModelForCausalLM.from_pretrained(checkpoint_path, **model_kwargs)
16
 
17
- new_model = "/content/checkpoint_dir/checkpoint-100" # change to the path where your model is saved
18
 
19
  model = PeftModel.from_pretrained(base_model, new_model)
20
  model = model.merge_and_unload()
 
14
  )
15
  base_model = AutoModelForCausalLM.from_pretrained(checkpoint_path, **model_kwargs)
16
 
17
+ new_model = "checkpoint_dir/checkpoint-100" # change to the path where your model is saved
18
 
19
  model = PeftModel.from_pretrained(base_model, new_model)
20
  model = model.merge_and_unload()