Skip to content

GitLab

"examples/llm/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "538b463025d2f971baa0e5b0328c1bab70599510"
Commit f4b1347b authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Add missing import

parent 9b861808
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
openfold/utils/checkpointing.py
View file @ f4b1347b
...@@ -11,6 +11,8 @@ ...@@ -11,6 +11,8 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import importlib
from typing import Any, Tuple, List, Callable, Optional
deepspeed_is_installed = importlib.util.find_spec("deepspeed") is not None deepspeed_is_installed = importlib.util.find_spec("deepspeed") is not None
if(deepspeed_is_installed): if(deepspeed_is_installed):
...@@ -18,7 +20,6 @@ if(deepspeed_is_installed): ...@@ -18,7 +20,6 @@ if(deepspeed_is_installed):
import torch import torch
import torch.utils.checkpoint import torch.utils.checkpoint
from typing import Any, Tuple, List, Callable, Optional
BLOCK_ARG = Any BLOCK_ARG = Any
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment