protenix
Browse files
protenix/openfold_local/utils/checkpointing.py
CHANGED
@@ -14,7 +14,8 @@
|
|
14 |
import importlib
|
15 |
from typing import Any, List, Callable, Optional
|
16 |
|
17 |
-
deepspeed_is_installed = importlib.util.find_spec("deepspeed") is not None
|
|
|
18 |
if deepspeed_is_installed:
|
19 |
import deepspeed
|
20 |
|
|
|
14 |
import importlib
|
15 |
from typing import Any, List, Callable, Optional
|
16 |
|
17 |
+
#deepspeed_is_installed = importlib.util.find_spec("deepspeed") is not None
|
18 |
+
deepspeed_is_installed = False
|
19 |
if deepspeed_is_installed:
|
20 |
import deepspeed
|
21 |
|