Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Oct 9, 2023

Commit

e095479

1 Parent(s): 215b416

get_input_embeddings() directly from base model

Files changed (1) hide show

qarac/models/QaracDecoderModel.py CHANGED Viewed

@@ -96,7 +96,7 @@ class QaracDecoderModel(transformers.RobertaModel,
         self.decoder_base = transformers.RobertaModel.from_pretrained(model_path,
                                                                       config=config)
         self.decoder_head = QaracDecoderHead(self.config,
-                                             self.decoder_base.roberta.get_input_embeddings())
         self.tokenizer = tokenizer

         self.decoder_base = transformers.RobertaModel.from_pretrained(model_path,
                                                                       config=config)
         self.decoder_head = QaracDecoderHead(self.config,
+                                             self.decoder_base.get_input_embeddings())
         self.tokenizer = tokenizer