Downloads · 30 days
2
3% of all-time downloads
doupari/llama3.1_8b_sft-solo-attn-v2-k24
llama3.1_8b_sft-solo-attn-v2-k24 is a machine learning model from doupari. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repo bundles LLOPA/TRI inference with minimal friction.
Downloads · 30 days
2
3% of all-time downloads
All-time downloads
64
Public
Repo size
64.3 GB
Likes
0
Public
Click a slice to open those files.
.bin32.1 GB · 50%
From the Hugging Face model README
This repo bundles LLOPA/TRI inference with minimal friction.
from transformers import AutoTokenizer, AutoModelForCausalLM
tok = AutoTokenizer.from_pretrained("your-repo")
model = AutoModelForCausalLM.from_pretrained("your-repo", trust_remote_code=True)
out = model.llopa_generate(
tokenizer=tok,
system="You are a helpful assistant.",
document="...",
question="...",
K=4,
prefill_mode="lower",
prefill_attn="causal",
)
print(out)