trim whitespace
Browse files- modeling_baichuan.py +1 -1
modeling_baichuan.py
CHANGED
@@ -607,7 +607,7 @@ class BaichuanForCausalLM(BaichuanPreTrainedModel):
|
|
607 |
) -> dict:
|
608 |
if past_key_values:
|
609 |
input_ids = input_ids[:, -1:]
|
610 |
-
|
611 |
# the cache may be in the standard format (e.g. in contrastive search)
|
612 |
if past_key_values[0][0].shape[0] == input_ids.shape[0]:
|
613 |
past_key_values = self._convert_to_baichuan_cache(past_key_values)
|
|
|
607 |
) -> dict:
|
608 |
if past_key_values:
|
609 |
input_ids = input_ids[:, -1:]
|
610 |
+
|
611 |
# the cache may be in the standard format (e.g. in contrastive search)
|
612 |
if past_key_values[0][0].shape[0] == input_ids.shape[0]:
|
613 |
past_key_values = self._convert_to_baichuan_cache(past_key_values)
|