Upload modeling_cpmbee.py
Browse files- modeling_cpmbee.py +0 -1
modeling_cpmbee.py
CHANGED
|
@@ -1910,7 +1910,6 @@ class CpmBeeForCausalLM(CpmBeePreTrainedModel):
|
|
| 1910 |
input_encoded.update(kwargs)
|
| 1911 |
input_encoded["generation_config"] = generation_config
|
| 1912 |
input_encoded["vocab_size"] = tokenizer.vocab_size
|
| 1913 |
-
print(tokenizer.vocab_size)
|
| 1914 |
|
| 1915 |
decode_res = self._generate(**input_encoded)
|
| 1916 |
|
|
|
|
| 1910 |
input_encoded.update(kwargs)
|
| 1911 |
input_encoded["generation_config"] = generation_config
|
| 1912 |
input_encoded["vocab_size"] = tokenizer.vocab_size
|
|
|
|
| 1913 |
|
| 1914 |
decode_res = self._generate(**input_encoded)
|
| 1915 |
|