Views
No views yet
from transformers import AutoModelForCausalLM, AutoTokenizer
repo_id = 'Monor/Qwen1.5-0.5B-h-world'
tokenizer = AutoTokenizer.from_pretrained(repo_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(repo_id, device_map="auto", trust_remote_code=True).eval()
inputs = tokenizer('小青突然来到了我的房间', return_tensors='pt')
inputs = inputs.to(model.device)
pred = model.generate(**inputs, max_new_tokens=32, num_beams=4, do_sample=True, pad_token_id=tokenizer.eos_token_id)
print(tokenizer.decode(pred.cpu()[0], skip_special_tokens=True))输出：小青突然来到了我的房间，吓了我一跳。她见我睡着了，就脱了衣服钻进了被窝里，然后搂着我睡着了。
| Model | Score |
|---|---|
| Human | 80.2 |
| rwkv-5-h-world-7B | 60.3 |
| rwkv-5-h-world-3B | 59.4 |
| rwkv-5-h-world-1b5 | 59.1 |
| Qwen1.5-0.5B-h-world | 57.9 |
| Yi-34B | 54.7 |
| rwkv-h-world-1b5 | 54.1 |
| rwkv-v4-7b-dengh | 50.0 |
| Yi-6B | 48.7 |
| Yi-34B-Chat-4bits | 48.1 |
| rwkv-h-world-0.4b | 46.8 |
| deepsex-34b | 45.9 |
| NSFW_13B_sft | 44.3 |
| CausalLM-14B-GPTQ | 43.4 |
| Baichuan2-7B-Base | 42.7 |
| RWKV-5-World-3B-v2-20231113-ctx4096 | 42.5 |
| rwkv-h-1b5 | 42.1 |
| RWKV-v5-12B-one-state-chat-16k | 41.3 |
| chatglm3-6b-base | 41.2 |
| RWKV-claude-4-World-7B-20230805-ctx65k | 40.2 |
| Baichuan2-13B-Base | 39.9 |
| RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096 | 39.3 |
| Baichuan2-13B-Chat-4bits | 37.4 |
| RWKV-5-World-1B5-v2-20231025-ctx4096 | 36.1 |
| Qwen-7B | 33.0 |
| chatglm3-6b | 30.5 |
| RWKV-4-World-CHNtuned-1.5B-v1-20230620-ctx4096 | 28.9 |
| RWKV-4-World-CHNtuned-0.4B-v1-20230618-ctx4096 | 22.9 |
| RWKV-4-Novel-3B-v1-Chn-20230412-ctx4096 | 20.4 |