| from __future__ import annotations | |
| from typing import TYPE_CHECKING | |
| if TYPE_CHECKING: | |
| import torch | |
| unsafe_torch_load = torch.load | |
| else: | |
| from modules.safe import unsafe_torch_load | |
| from __future__ import annotations | |
| from typing import TYPE_CHECKING | |
| if TYPE_CHECKING: | |
| import torch | |
| unsafe_torch_load = torch.load | |
| else: | |
| from modules.safe import unsafe_torch_load | |