| """ | |
| ISNet Background Remover Model | |
| This package provides the ISNet model for background removal using Transformers. | |
| """ | |
| from .modeling_isnet import ISNetConfig, ISNetForImageSegmentation | |
| # Register the model with transformers | |
| from transformers.models.auto.modeling_auto import MODEL_FOR_IMAGE_SEGMENTATION_MAPPING | |
| MODEL_FOR_IMAGE_SEGMENTATION_MAPPING.register(ISNetConfig, ISNetForImageSegmentation) | |
| __all__ = ["ISNetConfig", "ISNetForImageSegmentation"] |