Spaces:
Running
on
Zero
Running
on
Zero
Merge branch #happzy2633/open-o1' into 'DeFactOfficial/o3'
Browse files
V3.py
CHANGED
|
@@ -2,7 +2,7 @@ import spaces
|
|
| 2 |
|
| 3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 4 |
|
| 5 |
-
model_name = "
|
| 6 |
model = AutoModelForCausalLM.from_pretrained(
|
| 7 |
model_name,
|
| 8 |
torch_dtype="auto",
|
|
|
|
| 2 |
|
| 3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 4 |
|
| 5 |
+
model_name = "O1-OPEN/OpenO1-LLama-8B-v0.1"
|
| 6 |
model = AutoModelForCausalLM.from_pretrained(
|
| 7 |
model_name,
|
| 8 |
torch_dtype="auto",
|