Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
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)
|