Files
2023-10-01 16:54:56 +08:00

10 lines
650 B
Python

from .LamaRemove import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
from .saicinpainting.training.data.datasets import make_constant_area_crop_params
from .saicinpainting.training.losses.distance_weighting import make_mask_distance_weighter
from .saicinpainting.training.losses.feature_matching import feature_matching_loss, masked_l1_loss
from .saicinpainting.training.modules.fake_fakes import FakeFakesGenerator
from .saicinpainting.training.trainers.base import BaseInpaintingTrainingModule, make_multiscale_noise
from .saicinpainting.utils import add_prefix_to_keys, get_ramp
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']