|
--- |
|
license: apache-2.0 |
|
language: |
|
- zh |
|
pipeline_tag: text-generation |
|
tags: |
|
- not-for-all-audiences |
|
--- |
|
|
|
数据来源自[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) |
|
print(tokenizer.decode(pred.cpu()[0], skip_special_tokens=True)) |
|
|
|
输出:小青突然来到了我的房间,吓了我一跳。她见我睡着了,就脱了衣服钻进了被窝里,然后搂着我睡着了。 |
|
``` |