Spaces:

nkigumnov
/

banks-ethics-sentiment

Build error

nkigumnov commited on Apr 17, 2023

Commit

27e6827

•

1 Parent(s): 4dba23e

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -46,7 +46,7 @@ class UnionModel(nn.Module):
             param.requires_grad = False
         self.bert_model.pooler = nn.Linear(in_features=768, out_features=256)
-        self.bert_model.to('cuda')
         self.category_head = CategoryHead()
         self.sentiment_head = SentimentHead()

             param.requires_grad = False
         self.bert_model.pooler = nn.Linear(in_features=768, out_features=256)
+        self.bert_model.to('cpu')
         self.category_head = CategoryHead()
         self.sentiment_head = SentimentHead()