Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Sep 22, 2023

Commit

c833bc7

1 Parent(s): e09ba59

Fix typo

Files changed (1) hide show

qarac/models/QaracDecoderModel.py CHANGED Viewed

@@ -30,7 +30,7 @@ class QaracDecoderHead(keras.layers.Layer):
         self.concat = keras.layers.Concatenate(axis=1)
         self.layer_0 = transformers.models.roberta.modeling_tf_roberta.TFRobertaLayer(config)
         self.layer_1 = transformers.models.roberta.modeling_tf_roberta.TFRobertaLayer(config)
-        self.head = transformers.models.roberta.modeling_td_roberta.TFRobertaLMHead(config)
     def build(self,input_shape):
         """

         self.concat = keras.layers.Concatenate(axis=1)
         self.layer_0 = transformers.models.roberta.modeling_tf_roberta.TFRobertaLayer(config)
         self.layer_1 = transformers.models.roberta.modeling_tf_roberta.TFRobertaLayer(config)
+        self.head = transformers.models.roberta.modeling_tf_roberta.TFRobertaLMHead(config)
     def build(self,input_shape):
         """