---
language:
- zh
license: apache-2.0
tags:
- not-for-all-audiences
- nsfw
pipeline_tag: text-generation
---

# 18禁小说生成模型

数据来源自[h-corpus-2023](https://huggingface.co/datasets/a686d380/h-corpus-2023)  
模型来源自[Qwen1.5-0.5B](https://huggingface.co/Qwen/Qwen1.5-0.5B)

本模型仅用于学术研究，严禁商用。

```
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))
```

>输出：小青突然来到了我的房间，吓了我一跳。她见我睡着了，就脱了衣服钻进了被窝里，然后搂着我睡着了。

---
## WebUI已开发完成
[18禁小说生成器，点击跳转到GitHub](https://github.com/huangxinping/Qwen1.5-0.5B-h-world-webui)

---
## 评测得分
根据[h-eval](https://huggingface.co/datasets/a686d380/h-eval)评测方法，要求模型正确续写下一个单词，该模型获得57.9分。   
```评测过程使用h-eval中提供数据集，仅修改benchmark代码以适用Qwen1.5模型```

| 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|


---
## Doing：

**✌️基于qwen1.5-7B和qwen1.5-14B模型已训练到57%✌️**

*模型训练动机：科学研究*