Skip to content

Commit 4caafe0

Browse files
committed
Use optional import
1 parent a6bb925 commit 4caafe0

File tree

1 file changed

+3
-14
lines changed

1 file changed

+3
-14
lines changed

monai/networks/nets/autoencoderkl.py

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
from __future__ import annotations
1313

14-
import importlib.util
1514
import math
1615
from collections.abc import Sequence
1716
from typing import List
@@ -21,21 +20,11 @@
2120
import torch.nn.functional as F
2221

2322
from monai.networks.blocks import Convolution
24-
from monai.utils import ensure_tuple_rep
2523

2624
# To install xformers, use pip install xformers==0.0.16rc401
27-
if importlib.util.find_spec("xformers") is not None:
28-
import xformers
29-
import xformers.ops
30-
31-
has_xformers = True
32-
else:
33-
xformers = None
34-
has_xformers = False
35-
36-
# TODO: Use MONAI's optional_import
37-
# from monai.utils import optional_import
38-
# xformers, has_xformers = optional_import("xformers.ops", name="xformers")
25+
from monai.utils import ensure_tuple_rep, optional_import
26+
27+
xformers, has_xformers = optional_import("xformers")
3928

4029
__all__ = ["AutoencoderKL"]
4130

0 commit comments

Comments
 (0)