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