Downloads · 30 days
274
21% of all-time downloads
bloom-396m-chat
WangZeJun/bloom-396m-chat
bloom-396m-chat is a text generation model from WangZeJun. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as bigscience-bloom-rail-1.0.
https://github.com/zejunwang1/bloomtuning
All-time downloads
1.3K
Public
Parameters
350M
2.8 GB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin1.4 GB · 50%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- bigscience-bloom-rail-1.0
- Model type
- bloom
- Access
- Public
- Created
- May 23, 2023
- Updated
- Aug 29, 2023
- SHA
- 39bbe997
Try a prompt
Model card
From the Hugging Face model README
https://github.com/zejunwang1/bloom_tuning
可以通过如下代码调用 bloom-396m-chat 模型来生成对话：
from transformers import BloomTokenizerFast, BloomForCausalLM
model_name_or_path = "WangZeJun/bloom-396m-chat"
tokenizer = BloomTokenizerFast.from_pretrained(model_name_or_path)
model = BloomForCausalLM.from_pretrained(model_name_or_path).cuda()
model = model.eval()
input_pattern = "{}</s>"
text = "你好"
input_ids = tokenizer(input_pattern.format(text), return_tensors="pt").input_ids
input_ids = input_ids.cuda()
outputs = model.generate(input_ids, do_sample=True, max_new_tokens=1024, top_p=0.85,
temperature=0.3, repetition_penalty=1.2, eos_token_id=tokenizer.eos_token_id)
input_ids_len = input_ids.size(1)
response_ids = outputs[0][input_ids_len:]
response = tokenizer.decode(response_ids)
print(response)
- Task
- Text Generation
- Library
- transformers
- Type
- bloom
- License
- bigscience-bloom-rail-1.0
- Created
- May 23, 2023
- Updated
- Aug 29, 2023