Model Tested: bigscience/bloom-1b1
How I loaded the model in Colab:
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "bigscience/bloom-1b1"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
input_text = "The capital of France is"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=20)… See the full description on the dataset page:
https://huggingface.co/datasets/mohsul2000/bloom-1b1-blind-spots.