npvinHnivqn commited on
Commit
344d2d9
·
1 Parent(s): 143be44

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -0
README.md ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ ## Quick start
5
+ ```python
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer
7
+
8
+ model = AutoModelForCausalLM.from_pretrained("npvinHnivqn/phi-1_5-CRL-v0.2", trust_remote_code=True)
9
+ tokenizer = AutoTokenizer.from_pretrained("npvinHnivqn/phi-1_5-CRL-v0.2", trust_remote_code=True)
10
+
11
+ inputs = tokenizer('''<|USER|> Write a paragragh about animal''', return_tensors="pt", return_attention_mask=False)
12
+
13
+ outputs = model.generate(**inputs, max_length=200)
14
+ text = tokenizer.batch_decode(outputs)[0]
15
+ print(text)
16
+ ```