Update README.md
Browse files
README.md
CHANGED
@@ -34,11 +34,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
34 |
device = "cuda" # the device to load the model onto
|
35 |
|
36 |
model = AutoModelForCausalLM.from_pretrained(
|
37 |
-
"
|
38 |
torch_dtype="auto",
|
39 |
device_map="auto"
|
40 |
)
|
41 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
42 |
|
43 |
prompt = "Imagine you're helping someone who is feeling overwhelmed. How do you feel in this situation?"
|
44 |
messages = [
|
|
|
34 |
device = "cuda" # the device to load the model onto
|
35 |
|
36 |
model = AutoModelForCausalLM.from_pretrained(
|
37 |
+
"OEvortex/EMO-1.5B",
|
38 |
torch_dtype="auto",
|
39 |
device_map="auto"
|
40 |
)
|
41 |
+
tokenizer = AutoTokenizer.from_pretrained("OEvortex/EMO-1.5B")
|
42 |
|
43 |
prompt = "Imagine you're helping someone who is feeling overwhelmed. How do you feel in this situation?"
|
44 |
messages = [
|