w601sxs commited on
Commit
ecfed14
1 Parent(s): d46ec3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from datasets import load_dataset
6
  from trl import SFTTrainer
7
 
8
  ref_model = AutoModelForCausalLM.from_pretrained("w601sxs/b1ade-1b", torch_dtype=torch.bfloat16)
9
- peft_model_id = "w601sxs/b1ade-1b-orca-chkpt-230k"
10
 
11
  config = PeftConfig.from_pretrained(peft_model_id)
12
  model = PeftModel.from_pretrained(ref_model, peft_model_id)
 
6
  from trl import SFTTrainer
7
 
8
  ref_model = AutoModelForCausalLM.from_pretrained("w601sxs/b1ade-1b", torch_dtype=torch.bfloat16)
9
+ peft_model_id = "w601sxs/b1ade-1b-orca-chkpt-506k"
10
 
11
  config = PeftConfig.from_pretrained(peft_model_id)
12
  model = PeftModel.from_pretrained(ref_model, peft_model_id)