Abhaykoul commited on
Commit
095c281
1 Parent(s): eb5835a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
- "Abhaykoul/EMO-1B",
38
  torch_dtype="auto",
39
  device_map="auto"
40
  )
41
- tokenizer = AutoTokenizer.from_pretrained("Abhaykoul/EMO-1B")
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 = [