hiyouga commited on
Commit
c9d4975
1 Parent(s): 478dc90

trim whitespace

Browse files
Files changed (1) hide show
  1. 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)