Realluke commited on
Commit
db6d704
1 Parent(s): 50811df

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -15,8 +15,8 @@ Example:
15
  ```python
16
  from transformers import AutoTokenizer, AutoModelForCausalLM
17
 
18
- model = AutoModelForCausalLM.from_pretrained("Realluke/phi-2-senator-tweets", low_cpu_mem_usage=True, torch_dtype="auto", trust_remote_code=True)
19
- tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", cache_dir="./tokenizer", use_fast=True)
20
 
21
  inputs = tokenizer("[start]", return_tensors="pt", return_attention_mask=False)
22
  outputs = model.generate(**inputs, max_length=200)
 
15
  ```python
16
  from transformers import AutoTokenizer, AutoModelForCausalLM
17
 
18
+ model = AutoModelForCausalLM.from_pretrained("Realluke/phi-2-senator-tweets", torch_dtype="auto", trust_remote_code=True)
19
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True)
20
 
21
  inputs = tokenizer("[start]", return_tensors="pt", return_attention_mask=False)
22
  outputs = model.generate(**inputs, max_length=200)