ydshieh commited on
Commit
753f724
1 Parent(s): b5b1e4e

initial commit

Browse files
Files changed (1) hide show
  1. modeling_tf_my_model.py +3 -0
modeling_tf_my_model.py CHANGED
@@ -30,6 +30,9 @@ class TFMyModel(TFMyModelPretrainedModel):
30
  def call(
31
  self,
32
  hidden,
 
 
 
33
  ):
34
  breakpoint()
35
  self.linear(hidden)
 
30
  def call(
31
  self,
32
  hidden,
33
+ output_attentions=False,
34
+ output_hidden_states=False,
35
+ return_dict=False,
36
  ):
37
  breakpoint()
38
  self.linear(hidden)