Spaces:

PlayfulTechnology
/

QARAC

Build error

PeteBleackley commited on Apr 5, 2024

Commit

3fc3a1b

1 Parent(s): e973c2b

Diagnostics

Files changed (1) hide show

qarac/models/QaracTrainerModel.py CHANGED Viewed

@@ -102,5 +102,5 @@ class QaracTrainerModel(torch.nn.Module):
     def devices(self):
         print("Question encoder",self.question_encoder.device)
-        print('Answer Encoder',self.answer.encoder.device)
         print('Decoer',self.decoder.device)

     def devices(self):
         print("Question encoder",self.question_encoder.device)
+        print('Answer Encoder',self.answer_encoder.device)
         print('Decoer',self.decoder.device)