coh mech

ccl_win.py

@@ -125,7 +125,7 @@ class ccl_win(evaluate.Measurement):
             MODEL_CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/"
         
         if device is not None:
-            assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."
+            # assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."
             if device == "gpu":
                 device = "cuda"
         else: