For research on cognitive biases in LLMs. Used to test causal impact of pretraining vs instruction tuning.
Do not use in production, sensitive domains, or decision-critical applications.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("itay1itzhak/T5-Flan")
4tokenizer = AutoTokenizer.from_pretrained("itay1itzhak/T5-Flan")
5
6inputs = tokenizer("Example input?", return_tensors="pt")
7outputs = model.generate(**inputs)
8print(tokenizer.decode(outputs[0]))