Modelpublic

Langboat/bloom-389m-zh

sourceHugging Facebigscience-bloom-rail-1.0updated 4y agoView on Hugging Face

8likes1.5kdownloads

Model Card

This model is based on bigscience/bloom-560m.

We pruned its vocabulary from 250880 to 42437 with Chinese corpus to reduce GPU memory usage. So the total parameter is 389m now.

How to use

python

from transformers import BloomTokenizerFast, BloomForCausalLM

tokenizer = BloomTokenizerFast.from_pretrained('Langboat/bloom-389m-zh')
model = BloomForCausalLM.from_pretrained('Langboat/bloom-389m-zh')

print(tokenizer.batch_decode(model.generate(tokenizer.encode('中国的首都是', return_tensors='pt'))))