Spaces:

nickil
/

weakly-supervised-parsing

Build error

nickil commited on Apr 19, 2022

Commit

38c2a22

•

1 Parent(s): b0ceaae

update app

Files changed (1) hide show

weakly_supervised_parser/model/trainer.py CHANGED Viewed

@@ -99,7 +99,7 @@ class InsideOutsideStringClassifier:
         )
     def load_model(self, pre_trained_model_path):
-        self.model = InferenceSession(pre_trained_model_path, providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path, use_fast=True)
     def preprocess_function(self, data):

         )
     def load_model(self, pre_trained_model_path):
+        self.model = InferenceSession(pre_trained_model_path, providers=["CPUExecutionProvider"]) #providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path, use_fast=True)
     def preprocess_function(self, data):