Spaces:

nkigumnov
/

banks-ethics-sentiment

Build error

nkigumnov commited on Apr 17, 2023

Commit

4dba23e

1 Parent(s): c589743

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -40,7 +40,7 @@ class UnionModel(nn.Module):
     def __init__(self, model_path):
         super(UnionModel, self).__init__()
-        self.bert_model = None #AutoModel.from_pretrained(model_path)
         for _, param in self.bert_model.named_parameters():
             param.requires_grad = False

     def __init__(self, model_path):
         super(UnionModel, self).__init__()
+        self.bert_model = AutoModel.from_pretrained(model_path)
         for _, param in self.bert_model.named_parameters():
             param.requires_grad = False