Views
No views yet
1from transformers import BloomTokenizerFast, BloomForCausalLM
2
3tokenizer = BloomTokenizerFast.from_pretrained('Langboat/bloom-2b5-zh')
4model = BloomForCausalLM.from_pretrained('Langboat/bloom-2b5-zh')
5
6print(tokenizer.batch_decode(model.generate(tokenizer.encode('中国的首都是', return_tensors='pt'))))