Downloads · 30 days
4
12% of all-time downloads
Dhruvil47/falcon-7b-bioarxiv-ckpt-3.5
falcon-7b-bioarxiv-ckpt-3.5 is a machine learning model from Dhruvil47. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft.
Downloads · 30 days
4
12% of all-time downloads
All-time downloads
33
Public
Repo size
1.6 GB
Likes
0
Public
Click a slice to open those files.
.pt1 GB · 67%
From the Hugging Face model README
from transformers import AutoTokenizer, AutoModelForCausalLM
import transformers
import torch
model = "Dhruvil47/falcon-7b-bioarxiv-ckpt-3.5"
tokenizer = AutoTokenizer.from_pretrained(model)
pipeline = transformers.pipeline(
"text-generation",
model=model,
tokenizer=tokenizer,
torch_dtype=torch.bfloat16,
trust_remote_code=True,
device_map="auto",
)
sequences = pipeline(
"NKX2-1 works through",
max_length=200,
do_sample=True,
top_k=10,
num_return_sequences=1,
eos_token_id=tokenizer.eos_token_id,
)
for seq in sequences:
print(f"Result: {seq['generated_text']}")