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 |
|