Kowsher commited on
Commit
f14f38b
1 Parent(s): 092918d

Update ChatFalcon.py

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