caedencode
/

Caeden-o1

Text Generation

chain-of-thought

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

caedencode commited on Nov 30, 2024

Commit

094c828

·

verified ·

1 Parent(s): 1edd18f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("caedencode/Caeden-o1")
-tokenizer = AutoTokenizer.from_pretrained("caedencode/Caeden-o1")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)

 from transformers import AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("caedencode/Caeden-o1")
+tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-1.5B")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)