RTM's picture
Update s.py
a8f2e52
raw
history blame contribute delete
189 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("satvikag/chatbot")
model = AutoModelForCausalLM.from_pretrained("satvikag/chatbot")