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
|