playdev7 commited on
Commit
115562b
·
verified ·
1 Parent(s): b3ef5e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -19,10 +19,10 @@ library_name: transformers
19
 
20
  ## Model Generation
21
  ```
22
- from transforemrs import AutoTokenizer, AutoModelForCausalLM
23
-
24
- model = AutoModelForCausalLM.from_pretrained("playdev7/theseed", device_map="auto")
25
- tokenizer = AutoTokenizer.from_pretrained("playdev7/theseed", use_fast=True)
26
  ```
27
  ---
28
  based on Mixtral
 
19
 
20
  ## Model Generation
21
  ```
22
+ from transformers import AutoConfig, AutoModel, AutoTokenizer
23
+ config = AutoConfig.from_pretrained("playdev7/theseed", revision=revision)
24
+ model = AutoModel.from_pretrained("playdev7/theseed", revision=revision, device_map="auto")
25
+ tokenizer = AutoTokenizer.from_pretrained("playdev7/theseed", revision=revision)
26
  ```
27
  ---
28
  based on Mixtral