toto10's picture
c1a010c78919f8efef07b18490f90b54ce38605a218a48907aa972358dd0fbc1
9c5ffc0
raw
history blame
194 Bytes
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