Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Oct 9, 2023

Commit

14d83dc

1 Parent(s): 4d8fdac

Correct dimension of consistency cosine

Files changed (1) hide show

qarac/models/QaracTrainerModel.py CHANGED Viewed

@@ -39,7 +39,7 @@ class QaracTrainerModel(torch.nn.Module):
         self.decoder = qarac.models.QaracDecoderModel.QaracDecoderModel(base_model_path,
                                                                         config,
                                                                         tokenizer)
-        self.cosine = torch.nn.CosineSimilarity(dim=2,eps=1.0e-12)
     def forward(self,
                 all_text,

         self.decoder = qarac.models.QaracDecoderModel.QaracDecoderModel(base_model_path,
                                                                         config,
                                                                         tokenizer)
+        self.cosine = torch.nn.CosineSimilarity(dim=1,eps=1.0e-12)
     def forward(self,
                 all_text,