Langboat
/

bloom-1b4-zh

Text Generation

text-generation-inference

Model card Files Files and versions

wangyulong commited on Aug 31, 2022

Commit

73d95d9

·

1 Parent(s): 9ccfebc

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -16,8 +16,8 @@ We pruned its vocabulary from 250880 to 46145 with Chinese corpus to reduce GPU
 ```python
 from transformers import BloomTokenizerFast, BloomForCausalLM
-tokenizer = BloomTokenizerFast.from_pretrained('Langboat/bloom-1.4b-zh')
-model = BloomForCausalLM.from_pretrained('Langboat/bloom-1.4b-zh')
 print(tokenizer.batch_decode(model.generate(tokenizer.encode('中国的首都是', return_tensors='pt'))))
 ```

 ```python
 from transformers import BloomTokenizerFast, BloomForCausalLM
+tokenizer = BloomTokenizerFast.from_pretrained('Langboat/bloom-1b4-zh')
+model = BloomForCausalLM.from_pretrained('Langboat/bloom-1b4-zh')
 print(tokenizer.batch_decode(model.generate(tokenizer.encode('中国的首都是', return_tensors='pt'))))
 ```