Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
34 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
35 |
|
36 |
# Model name
|
37 |
-
model_name = "
|
38 |
|
39 |
# Load tokenizer and model
|
40 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
34 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
35 |
|
36 |
# Model name
|
37 |
+
model_name = "mxmax/Llama3-8B-Chat-Dpo"
|
38 |
|
39 |
# Load tokenizer and model
|
40 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|