RWKV-7 "Goose" with Expressive Dynamic State Evolution
Paper • 2503.14456 • Published • 154
How to use d3banjan/RWKV7-Goose-World3-2.9B-HF with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="d3banjan/RWKV7-Goose-World3-2.9B-HF", trust_remote_code=True)
messages = [
{"role": "user", "content": "Who are you?"},
]
pipe(messages) # Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("d3banjan/RWKV7-Goose-World3-2.9B-HF", trust_remote_code=True, dtype="auto")How to use d3banjan/RWKV7-Goose-World3-2.9B-HF with vLLM:
# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "d3banjan/RWKV7-Goose-World3-2.9B-HF"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "d3banjan/RWKV7-Goose-World3-2.9B-HF",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'docker model run hf.co/d3banjan/RWKV7-Goose-World3-2.9B-HF
How to use d3banjan/RWKV7-Goose-World3-2.9B-HF with SGLang:
# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
--model-path "d3banjan/RWKV7-Goose-World3-2.9B-HF" \
--host 0.0.0.0 \
--port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "d3banjan/RWKV7-Goose-World3-2.9B-HF",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'docker run --gpus all \
--shm-size 32g \
-p 30000:30000 \
-v ~/.cache/huggingface:/root/.cache/huggingface \
--env "HF_TOKEN=<secret>" \
--ipc=host \
lmsysorg/sglang:latest \
python3 -m sglang.launch_server \
--model-path "d3banjan/RWKV7-Goose-World3-2.9B-HF" \
--host 0.0.0.0 \
--port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "d3banjan/RWKV7-Goose-World3-2.9B-HF",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'How to use d3banjan/RWKV7-Goose-World3-2.9B-HF with Docker Model Runner:
docker model run hf.co/d3banjan/RWKV7-Goose-World3-2.9B-HF
This is RWKV-7 model under flash-linear attention format.
Install flash-linear-attention and the latest version of transformers before using this model:
pip install flash-linear-attention==0.3.0
pip install 'transformers>=4.48.0'
You can use this model just as any other HuggingFace models:
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained('fla-hub/rwkv7-2.9B-world', trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained('fla-hub/rwkv7-2.9B-world', trust_remote_code=True)
model = model.cuda() # Supported on Nvidia/AMD/Intel eg. model.xpu()
prompt = "What is a large language model?"
messages = [
{"role": "user", "content": prompt}
]
text = tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True
)
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
generated_ids = model.generate(
**model_inputs,
max_new_tokens=4096,
do_sample=True,
temperature=1.0,
top_p=0.3,
repetition_penalty=1.2
)
generated_ids = [
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
]
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=False)[0]
print(response)
This model is trained on the World v3 with a total of 3.119 trillion tokens.
Q: safetensors metadata is none.
A: upgrade transformers to >=4.48.0: pip install 'transformers>=4.48.0'
Base model
BlinkDL/rwkv-7-world