Files changed (1) hide show
  1. modeling_falcon.py +1 -0
modeling_falcon.py CHANGED
@@ -722,6 +722,7 @@ class FalconModel(FalconPreTrainedModel):
722
  if input_ids is not None and inputs_embeds is not None:
723
  raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
724
  elif input_ids is not None:
 
725
  batch_size, seq_length = input_ids.shape
726
  elif inputs_embeds is not None:
727
  batch_size, seq_length, _ = inputs_embeds.shape
 
722
  if input_ids is not None and inputs_embeds is not None:
723
  raise ValueError("You cannot specify both input_ids and inputs_embeds at the same time")
724
  elif input_ids is not None:
725
+ print(input_ids)
726
  batch_size, seq_length = input_ids.shape
727
  elif inputs_embeds is not None:
728
  batch_size, seq_length, _ = inputs_embeds.shape