Di Zhang
commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -29,8 +29,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 29 |
|
| 30 |
|
| 31 |
|
| 32 |
-
tokenizer = AutoTokenizer.from_pretrained("/
|
| 33 |
-
model = AutoModelForCausalLM.from_pretrained("/
|
| 34 |
|
| 35 |
|
| 36 |
|
|
|
|
| 29 |
|
| 30 |
|
| 31 |
|
| 32 |
+
tokenizer = AutoTokenizer.from_pretrained("SimpleBerry/LLaMA-O1-Supervised-1129")
|
| 33 |
+
model = AutoModelForCausalLM.from_pretrained("SimpleBerry/LLaMA-O1-Supervised-1129",device_map='auto')
|
| 34 |
|
| 35 |
|
| 36 |
|