caedencode commited on
Commit
efe5fc3
·
verified ·
1 Parent(s): 7927897

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("caedenai/fine-tuned")
56
- tokenizer = AutoTokenizer.from_pretrained("caedenai/fine-tuned")
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")