Views
No views yet

| Tasks | Metric | QRWKV-QwQ-32B | Qwen/QwQ-32B | QRWKV-72B | Qwen2.5-72B-Instruct |
|---|---|---|---|---|---|
| arc_challenge | acc_norm | 0.5640 | 0.5563 | 0.6382 | 0.6323 |
| arc_easy | acc_norm | 0.7837 | 0.7866 | 0.8443 | 0.8329 |
| hellaswag | acc_norm | 0.8303 | 0.8407 | 0.8573 | 0.8736 |
| lambada_openai | acc | 0.6621 | 0.6683 | 0.7539 | 0.7506 |
| piqa | acc | 0.8036 | 0.7976 | 0.8248 | 0.8357 |
| sciq | acc | 0.9630 | 0.9630 | 0.9670 | 0.9740 |
| winogrande | acc | 0.7324 | 0.7048 | 0.7956 | 0.7632 |
| mmlu | acc | 0.7431 | 0.7985 | 0.7746 | 0.8338 |
Note: All benchmarks except MMLU are 0-shot and Version 1. For MMLU, it's Version 2.
transformers1# ...
2
3model = AutoModelForCausalLM.from_pretrained("featherless-ai/QRWKV-72B", trust_remote_code=True)
4
5# ...1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "featherless-ai/QRWKV-72B"
4
5model = AutoModelForCausalLM.from_pretrained(
6 model_name,
7 torch_dtype="auto",
8 device_map="auto",
9 trust_remote_code=True,
10)
11
12tokenizer = AutoTokenizer.from_pretrained(model_name)
13
14prompt = """There is a very famous song that I recall by the singer's surname as Astley.
15 I can't remember the name or the youtube URL that people use to link as an example url.
16 What's song name?"""
17messages = [
18 {"role": "system", "content": "You are a helpful assistant."},
19 {"role": "user", "content": prompt},
20]
21text = tokenizer.apply_chat_template(
22 messages, tokenize=False, add_generation_prompt=True
23)
24model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
25
26generated_ids = model.generate(**model_inputs, max_new_tokens=512)
27generated_ids = [
28 output_ids[len(input_ids) :]
29 for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
30]
31
32response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]