AventIQ-AI
/

pythia-410m-chatbot

Model card Files Files and versions Community

ayushsinha commited on Feb 6

Commit

04c28a8

·

verified ·

1 Parent(s): c50c636

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ pip install transformers torch
 ```python
 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

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