Downloads · 30 days
0
laurapiano3/fuzzy
fuzzy is a machine learning model from laurapiano3. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
from transformers import BloomTokenizerFast, BloomForCausalLM
Downloads · 30 days
0
Access
Public
Updated Aug 2, 2024
Repo size
—
Likes
0
Public
Click a slice to open those files.
Other1.5 KB · 62%
From the Hugging Face model README
from transformers import BloomTokenizerFast, BloomForCausalLM
model_name = "bigscience/bloom-560m" # You can choose a different model size if needed tokenizer = BloomTokenizerFast.from_pretrained(model_name) model = BloomForCausalLM.from_pretrained(model_name)
prompt = "Once upon a time, in a land far away, there was a"
inputs = tokenizer(prompt, return_tensors="pt")
output = model.generate( inputs["input_ids"], max_length=100, # Adjust the length as needed num_return_sequences=1, pad_token_id=tokenizer.eos_token_id, top_k=50, # Adjust sampling parameters as needed top_p=0.95 )
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print("Generated Text:") print(generated_text)