Commit
·
2dd5eef
1
Parent(s):
295ca87
Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ from peft import PeftModel
|
|
14 |
base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
|
15 |
peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-base")
|
16 |
|
17 |
-
inputs = tokenizer("
|
18 |
outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)
|
19 |
print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
|
20 |
```
|
|
|
14 |
base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
|
15 |
peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-base")
|
16 |
|
17 |
+
inputs = tokenizer("List a few tips to get good scores in math.", return_tensors="pt")
|
18 |
outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)
|
19 |
print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
|
20 |
```
|