Spaces:

spark-ds549
/

Epik

Sleeping

Minh Q. Le commited on Dec 6, 2023

Commit

8cac419

1 Parent(s): efe5117

Changed COMET to using cpu

Files changed (1) hide show

Model/COSMIC/feature_extraction/comet/csk_feature_extract.py CHANGED Viewed

@@ -13,7 +13,7 @@ class CSKFeatureExtractor:
     def __init__(self, dir="."):
         super(CSKFeatureExtractor, self).__init__()
-        device = 0
         model_file = os.path.join(
             dir, "comet/pretrained_models/atomic_pretrained_model.pickle"
         )

     def __init__(self, dir="."):
         super(CSKFeatureExtractor, self).__init__()
+        device = "cpu"
         model_file = os.path.join(
             dir, "comet/pretrained_models/atomic_pretrained_model.pickle"
         )