MD-Mushfiqur123/BrewPilot-1.0

BrewPilot 1.0

BrewPilot 1.0 একটি এআই অ্যাসিস্ট্যান্ট, বাংলা ও ইংরেজি ভাষায় কথোপকথন ও সাধারণ কাজে সহায়তা করতে সক্ষম।

Developed by

MD-Mushfiqur123

ব্যবহার

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_name = "MD-Mushfiqur123/BrewPilot-1.0"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, dtype=torch.float16, device_map="auto")

messages = [{"role": "user", "content": [{"type": "text", "text": "তোমার নাম কি?"}]}]
inputs = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt", return_dict=True).to(model.device)
output = model.generate(**inputs, max_new_tokens=200)
print(tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))

License

MIT License.