openbmb
/

cpm-bee-5b

Text Generation

feature-extraction

Model card Files Files and versions Community

jeffreygo commited on Jun 16, 2023

Commit

e3314e1

•

1 Parent(s): f2b0065

Update modeling_cpmbee.py

Files changed (1) hide show

modeling_cpmbee.py +0 -1

modeling_cpmbee.py CHANGED Viewed

@@ -1910,7 +1910,6 @@ class CpmBeeForCausalLM(CpmBeePreTrainedModel):
         input_encoded.update(kwargs)
         input_encoded["generation_config"] = generation_config
         input_encoded["vocab_size"] = tokenizer.vocab_size
-        print(tokenizer.vocab_size)
         decode_res = self._generate(**input_encoded)

         input_encoded.update(kwargs)
         input_encoded["generation_config"] = generation_config
         input_encoded["vocab_size"] = tokenizer.vocab_size
         decode_res = self._generate(**input_encoded)