wsntxxn
/

effb2-trm-audiocaps-captioning

Feature Extraction

Model card Files Files and versions

wsntxxn commited on Jul 28

Commit

47c316d

·

verified ·

1 Parent(s): 8f3d31e

Update hf_wrapper.py

Files changed (1) hide show

hf_wrapper.py +1 -1

hf_wrapper.py CHANGED Viewed

@@ -1164,7 +1164,7 @@ class Effb2TrmCaptioningModel(PreTrainedModel):
         model = super().from_pretrained(
             pretrained_model_name_or_path, *args, **kwargs
         )
-        model.model.decoder.word_embedding.weight = model.model.decoder.classifier.weight
         return model
     def forward(self,

         model = super().from_pretrained(
             pretrained_model_name_or_path, *args, **kwargs
         )
+        model.model.model.decoder.word_embedding.weight = model.model.model.decoder.classifier.weight
         return model
     def forward(self,