yi-01-ai commited on
Commit
e00f7cb
1 Parent(s): a38ef3d

Auto Sync from git://github.com/01-ai/Yi.git/commit/231bf09e44204b105eeaab5f2aa5c50644544ace

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -321,8 +321,8 @@ Or if you'd like to get your hands dirty:
321
  ```python
322
  from transformers import AutoModelForCausalLM, AutoTokenizer
323
 
324
- model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-34B", device_map="auto", torch_dtype="auto", trust_remote_code=True)
325
- tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-34B", trust_remote_code=True)
326
  inputs = tokenizer("There's a place where time stands still. A place of breath taking wonder, but also", return_tensors="pt")
327
  max_length = 256
328
 
 
321
  ```python
322
  from transformers import AutoModelForCausalLM, AutoTokenizer
323
 
324
+ model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-34B", device_map="auto", torch_dtype="auto")
325
+ tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-34B")
326
  inputs = tokenizer("There's a place where time stands still. A place of breath taking wonder, but also", return_tensors="pt")
327
  max_length = 256
328