kwang2049
/

long-coref

Feature Extraction

endpoints-template

Model card Files Files and versions

kwang2049 commited on Feb 12, 2024

Commit

04d2050

·

1 Parent(s): 30adfc7

gpu or cpu

Files changed (1) hide show

pipeline.py +0 -1

pipeline.py CHANGED Viewed

@@ -15,7 +15,6 @@ class PreTrainedPipeline:
             config=Params.from_file(os.path.join(path, CHECKPOINT, "config.json")),
         )
         self.predictor = CorefPredictor.from_extracted_archive(archive_content)
-        self.predictor.set_device("cpu")
     def __call__(self, data: str) -> Dict[str, Any]:
         """

             config=Params.from_file(os.path.join(path, CHECKPOINT, "config.json")),
         )
         self.predictor = CorefPredictor.from_extracted_archive(archive_content)
     def __call__(self, data: str) -> Dict[str, Any]:
         """