protenix

protenix/openfold_local/model/primitives.py

@@ -19,11 +19,13 @@ from typing import Callable, List, Optional, Tuple
 
 import numpy as np
 
-deepspeed_is_installed = importlib.util.find_spec("deepspeed") is not None
-ds4s_is_installed = (
-    deepspeed_is_installed
-    and importlib.util.find_spec("deepspeed.ops.deepspeed4science") is not None
-)
+# deepspeed_is_installed = importlib.util.find_spec("deepspeed") is not None
+# ds4s_is_installed = (
+#     deepspeed_is_installed
+#     and importlib.util.find_spec("deepspeed.ops.deepspeed4science") is not None
+# )
+deepspeed_is_installed = False
+ds4s_is_installed = False
 if deepspeed_is_installed:
     import deepspeed