Update ChatFalcon.py
Browse files- ChatFalcon.py +4 -0
ChatFalcon.py
CHANGED
@@ -900,6 +900,10 @@ class FalconForCausalLM(FalconPreTrainedModel):
|
|
900 |
|
901 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
902 |
|
|
|
|
|
|
|
|
|
903 |
transformer_outputs = self.transformer(
|
904 |
input_ids,
|
905 |
token_type_ids = token_type_ids,
|
|
|
900 |
|
901 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
902 |
|
903 |
+
print('working')
|
904 |
+
print(input_ids)
|
905 |
+
print(token_type_ids)
|
906 |
+
|
907 |
transformer_outputs = self.transformer(
|
908 |
input_ids,
|
909 |
token_type_ids = token_type_ids,
|