metadata
license: apache-2.0
language:
- zh
pipeline_tag: text-generation
tags:
- not-for-all-audiences
数据来源自h-corpus-2023
模型来源自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))
输出：小青突然来到了我的房间，吓了我一跳。她见我睡着了，就脱了衣服钻进了被窝里，然后搂着我睡着了。