Spaces:

happzy2633
/

open-o1

Running on Zero

happzy2633 commited on 5 days ago

Commit

55535ff

•

1 Parent(s): 1dcc6a9

Update V3.py

Files changed (1) hide show

V3.py CHANGED Viewed

@@ -2,7 +2,7 @@ import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_name = "happzy2633/qwen2.5-7b-ins-v3"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype="auto",

 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_name = "O1-OPEN/OpenO1-LLama-8B-v0.1"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     torch_dtype="auto",