Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model = "Hulk810154/BABYAI"
4tokenizer = AutoTokenizer.from_pretrained(model)
5model = AutoModelForCausalLM.from_pretrained(model)
6
7# Example inference (text-based)
8inputs = tokenizer("What are my bills due this week?", return_tensors="pt")
9outputs = model.generate(**inputs, max_new_tokens=64)
10print(tokenizer.decode(outputs[0]))
11
12# Advanced: integrate with LangChain or custom workflow systems for RAG and memory.---
13license: apache-2.0
14---
15## Model Weights
16
17For now, BabyAI uses the weights from [Mistral 7B Instruct v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2).