AventIQ-AI
/

pythia-410m-chatbot

Model card Files Files and versions

ayushsinha commited on Feb 5

Commit

ff3c68d

·

verified ·

1 Parent(s): 0f7754d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ pip install transformers torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("EleutherAI/pythia-410m")
-model = AutoModelForCausalLM.from_pretrained("AventIQ-AI/pythia-410m")
 tokenizer.pad_token = tokenizer.eos_token

 from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("EleutherAI/pythia-410m")
+model = AutoModelForCausalLM.from_pretrained("AventIQ-AI/pythia-410m-chatbot")
 tokenizer.pad_token = tokenizer.eos_token