---
license: apache-2.0
datasets:
- BelleGroup/train_1M_CN
- BelleGroup/multiturn_chat_0.8M
- jeffwan/sharegpt_vicuna
language:
- zh
- en
library_name: transformers
pipeline_tag: text-generation
tags:
- chat
widget:
- text: "<Human>: Hello <eoh> <Assistant>:"
  example_title: "Hello"
- text: "<Human>: 你好 <eoh> <Assistant>:"
  example_title: "你好"
- text: "<Human>: What should I do if I can't sleep at night? <eoh> <Assistant>:"
  example_title: "insomnia"
- text: "<Human>: 晚上睡不着应该怎么办？ <eoh> <Assistant>:"
  example_title: "失眠"
inference:
  parameters:
    temperature: 0.8
    max_new_tokens: 128
---
# ChatBLOOM

ChatBLOOM是基于[BLOOM](https://huggingface.co/bigscience/bloom-1b7)（17亿参数）训练的中英双语对话语言模型，此模型为SFT版本。
详见[Github](https://github.com/NicholasCao/ChatBloom)。

ChatBLOOM is a Chinese-English bilingual dialogue language model trained based on [BLOOM](https://huggingface.co/bigscience/bloom-1b7) (1.7 billion parameters). This model is the SFT version.
See [Github](https://github.com/NicholasCao/ChatBloom) for details.

## Usage
```python
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained('nicholascao/chatbloom-1b7-sft')
tokenizer.pad_token_id = tokenizer.eos_token_id

model = AutoModelForCausalLM.from_pretrained('nicholascao/chatbloom-1b7-sft').half()

inputs = tokenizer('<Human>: Hello <eoh> <Assistant>:', return_tensors='pt').to(torch.cuda.current_device())
model.to(torch.cuda.current_device())

output = model.generate(**inputs, max_length=768, do_sample=True, temperature=0.8, top_k=50, early_stopping=True, repetition_penalty=1.1)
output = tokenizer.decode(output[0], skip_special_tokens=True)
print(output)
```

## Limitation and Usage Limits

我们使用的数据集(例如[BELLE](https://github.com/LianjiaTech/BELLE))要求开发人员仅将数据用于研究目的。
因此，不允许将我们的模型用于商业以及其他潜在的有害用途。

The datasets we used (e.g. [BELLE](https://github.com/LianjiaTech/BELLE)) require developers only use the data for research purposes. 
Thus, commercial and other potentially harmful uses of our models are not allowed.