Update README.md
Browse files
README.md
CHANGED
@@ -52,8 +52,8 @@ import torch
|
|
52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
53 |
|
54 |
# Load the model and tokenizer
|
55 |
-
model = AutoModelForCausalLM.from_pretrained("
|
56 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
57 |
|
58 |
# Use appropriate device
|
59 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
53 |
|
54 |
# Load the model and tokenizer
|
55 |
+
model = AutoModelForCausalLM.from_pretrained("caedencode/Caeden-o1")
|
56 |
+
tokenizer = AutoTokenizer.from_pretrained("caedencode/Caeden-o1")
|
57 |
|
58 |
# Use appropriate device
|
59 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|