Update README.md
Browse files
README.md
CHANGED
@@ -28,7 +28,7 @@ This model was trained using AutoTrain. For more information, please visit [Auto
|
|
28 |
|
29 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
30 |
|
31 |
-
model_path = "mrcuddle/Tiny-DarkLlama"
|
32 |
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
34 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
28 |
|
29 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
30 |
|
31 |
+
model_path = "mrcuddle/Tiny-DarkLlama-Chat"
|
32 |
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
34 |
model = AutoModelForCausalLM.from_pretrained(
|