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 = [
|