File size: 194 Bytes
34097e9 |
1 2 3 4 5 6 7 8 9 10 11 |
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
|