Update for langchain
#113
by
mbinkamran
- opened
- 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
|