AI Assingment.ipynb - Colab
AI Assingment.ipynb - Colab
model_name = "distilgpt2"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
Once upon a time in my life, I was always very focused on my work and the world, and I was always focused on my work and the world, and I was always focused on my work and the
model_name = "facebook/blenderbot-400M-distill"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
def chatbot(prompt):
"""Generate a chatbot response"""
inputs = tokenizer(prompt, return_tensors="pt")
output = model.generate(
**inputs,
max_length=100,
pad_token_id=tokenizer.eos_token_id
)
response = chatbot(user_input)
print("Chatbot:", response)
model_name = "tiiuae/falcon-7b-instruct"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
def chatbot(prompt):
"""Generate a chatbot response"""
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
output = model.generate(
**inputs,
max_length=200,
pad_token_id=tokenizer.eos_token_id,
do_sample=True,
temperature=0.7,
top_k=50,
top_p=0.9
)
response = chatbot(user_input)
print("Chatbot:", response)