reasonwang commited on
Commit
dc3db86
1 Parent(s): abacac1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -0
README.md CHANGED
@@ -1,3 +1,20 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+ Out repository [flan-alpaca-lora](https://github.com/Reason-Wang/flan-alpaca-lora) contains the details to train flan-t5 with [Alpaca](https://github.com/tatsu-lab/stanford_alpaca) instructions and [low-rank adaptation](https://arxiv.org/abs/2106.09685).
5
+
6
+ You can use the following code easily.
7
+
8
+ Usage:
9
+
10
+ ```python
11
+ import transformers
12
+ from peft import PeftModel
13
+
14
+ base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
15
+ peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-large")
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
+ ```