All modules for which code is available

• adm-g_ddim25_8xb32_imagenet_256x256
• adm-g_ddim25_8xb32_imagenet_512x512
• adm-g_ddim25_8xb32_imagenet_64x64
• adm_ddim250_8xb32_imagenet_256x256
• adm_ddim250_8xb32_imagenet_512x512
• adm_ddim250_8xb32_imagenet_64x64
• base_biggan_128x128
• base_cyclegan
• base_dcgan_128x128
• base_dcgan_64x64
• base_deepfillv1
• base_deepfillv2
• base_edvr
• base_gl
• base_glean
• base_liif
• base_pconv
• base_pix2pix
• base_sagan_128x128
• base_sagan_32x32
• base_sngan_proj_128x128
• base_sngan_proj_32x32
• base_styleganv1
• base_styleganv2
• base_styleganv3
• base_tof
• basicvsr_test_config
• biggan-deep_cvt-hugging-face-rgb_imagenet1k-128x128
• biggan-deep_cvt-hugging-face_rgb_imagenet1k-256x256
• biggan-deep_cvt-hugging-face_rgb_imagenet1k-512x512
• biggan_2xb25-500kiters_cifar10-32x32
• biggan_ajbrock-sn_8xb32-1500kiters_imagenet1k-128x128
• biggan_cvt-BigGAN-PyTorch-rgb_imagenet1k-128x128
• celeba
• cifar10_noaug
• cifar10_nopad
• comp1k
• deblurring-defocus_test_config
• deblurring-motion_test_config
• decompression_test_config
• default_runtime
• denoising-gaussian_color_test_config
• denoising-gaussian_gray_test_config
• denoising-real_test_config
• deraining_test_config
• dreambooth
• dreambooth-finetune_text_encoder
• dreambooth_lora
• eg3d_cvt-official-rgb_afhq-512x512
• eg3d_cvt-official-rgb_ffhq-512x512
• eg3d_cvt-official-rgb_shapenet-128x128
• ffhq_flip
• gen_default_runtime
• grow_scale_imgs_ffhq_styleganv1
• imagenet_128
• imagenet_256
• imagenet_512
• imagenet_64
• imagenet_noaug_128
• inpaint_default_runtime
• inst-colorizatioon_full_official_cocostuff-256x256
• liif_test_config
• lsun_stylegan
• matting_default_runtime
• mmagic.apis.inferencers.base_mmagic_inferencer
• mmagic.apis.inferencers.colorization_inferencer
• mmagic.apis.inferencers.conditional_inferencer
• mmagic.apis.inferencers.controlnet_animation_inferencer
• mmagic.apis.inferencers.diffusers_pipeline_inferencer
• mmagic.apis.inferencers.eg3d_inferencer
• mmagic.apis.inferencers.image_super_resolution_inferencer
• mmagic.apis.inferencers.inference_functions
• mmagic.apis.inferencers.inpainting_inferencer
• mmagic.apis.inferencers.matting_inferencer
• mmagic.apis.inferencers.text2image_inferencer
• mmagic.apis.inferencers.translation_inferencer
• mmagic.apis.inferencers.unconditional_inferencer
• mmagic.apis.inferencers.video_interpolation_inferencer
• mmagic.apis.inferencers.video_restoration_inferencer
• mmagic.apis.mmagic_inferencer
• mmagic.datasets.basic_conditional_dataset
• mmagic.datasets.basic_frames_dataset
• mmagic.datasets.basic_image_dataset
• mmagic.datasets.categories
• mmagic.datasets.cifar10_dataset
• mmagic.datasets.comp1k_dataset
• mmagic.datasets.controlnet_dataset
• mmagic.datasets.data_utils
• mmagic.datasets.dreambooth_dataset
• mmagic.datasets.grow_scale_image_dataset
• mmagic.datasets.imagenet_dataset
• mmagic.datasets.mscoco_dataset
• mmagic.datasets.paired_image_dataset
• mmagic.datasets.singan_dataset
• mmagic.datasets.textual_inversion_dataset
• mmagic.datasets.transforms.albu_function
• mmagic.datasets.transforms.albumentations
• mmagic.datasets.transforms.alpha
• mmagic.datasets.transforms.aug_frames
• mmagic.datasets.transforms.aug_pixel
• mmagic.datasets.transforms.aug_shape
• mmagic.datasets.transforms.blur_kernels
• mmagic.datasets.transforms.crop
• mmagic.datasets.transforms.fgbg
• mmagic.datasets.transforms.formatting
• mmagic.datasets.transforms.generate_assistant
• mmagic.datasets.transforms.generate_frame_indices
• mmagic.datasets.transforms.get_masked_image
• mmagic.datasets.transforms.loading
• mmagic.datasets.transforms.matlab_like_resize
• mmagic.datasets.transforms.normalization
• mmagic.datasets.transforms.random_degradations
• mmagic.datasets.transforms.random_down_sampling
• mmagic.datasets.transforms.trimap
• mmagic.datasets.transforms.values
• mmagic.datasets.unpaired_image_dataset
• mmagic.engine.hooks.ema
• mmagic.engine.hooks.iter_time_hook
• mmagic.engine.hooks.pggan_fetch_data_hook
• mmagic.engine.hooks.pickle_data_hook
• mmagic.engine.hooks.reduce_lr_scheduler_hook
• mmagic.engine.hooks.visualization_hook
• mmagic.engine.optimizers.multi_optimizer_constructor
• mmagic.engine.optimizers.pggan_optimizer_constructor
• mmagic.engine.optimizers.singan_optimizer_constructor
• mmagic.engine.runner.log_processor
• mmagic.engine.runner.loop_utils
• mmagic.engine.runner.multi_loops
• mmagic.engine.schedulers.linear_lr_scheduler_with_interval
• mmagic.engine.schedulers.reduce_lr_scheduler
• mmagic.evaluation.evaluator
• mmagic.evaluation.functional.fid_inception
• mmagic.evaluation.functional.gaussian_funcs
• mmagic.evaluation.functional.inception_utils
• mmagic.evaluation.metrics.base_gen_metric
• mmagic.evaluation.metrics.base_sample_wise_metric
• mmagic.evaluation.metrics.connectivity_error
• mmagic.evaluation.metrics.equivariance
• mmagic.evaluation.metrics.fid
• mmagic.evaluation.metrics.gradient_error
• mmagic.evaluation.metrics.inception_score
• mmagic.evaluation.metrics.mae
• mmagic.evaluation.metrics.matting_mse
• mmagic.evaluation.metrics.metrics_utils
• mmagic.evaluation.metrics.ms_ssim
• mmagic.evaluation.metrics.mse
• mmagic.evaluation.metrics.niqe
• mmagic.evaluation.metrics.ppl
• mmagic.evaluation.metrics.precision_and_recall
• mmagic.evaluation.metrics.psnr
• mmagic.evaluation.metrics.sad
• mmagic.evaluation.metrics.snr
• mmagic.evaluation.metrics.ssim
• mmagic.evaluation.metrics.swd
• mmagic.models.archs.all_gather_layer
• mmagic.models.archs.aspp
• mmagic.models.archs.attention_injection
• mmagic.models.archs.downsample
• mmagic.models.archs.ensemble
• mmagic.models.archs.gated_conv_module
• mmagic.models.archs.img_normalize
• mmagic.models.archs.linear_module
• mmagic.models.archs.lora
• mmagic.models.archs.multi_layer_disc
• mmagic.models.archs.patch_disc
• mmagic.models.archs.resnet
• mmagic.models.archs.separable_conv_module
• mmagic.models.archs.simple_encoder_decoder
• mmagic.models.archs.smpatch_disc
• mmagic.models.archs.sr_backbone
• mmagic.models.archs.tokenizer
• mmagic.models.archs.upsample
• mmagic.models.archs.vgg
• mmagic.models.archs.wrapper
• mmagic.models.base_models.average_model
• mmagic.models.base_models.base_conditional_gan
• mmagic.models.base_models.base_edit_model
• mmagic.models.base_models.base_gan
• mmagic.models.base_models.base_mattor
• mmagic.models.base_models.base_translation_model
• mmagic.models.base_models.basic_interpolator
• mmagic.models.base_models.one_stage
• mmagic.models.base_models.two_stage
• mmagic.models.data_preprocessors.data_preprocessor
• mmagic.models.data_preprocessors.mattor_preprocessor
• mmagic.models.diffusion_schedulers.ddim_scheduler
• mmagic.models.diffusion_schedulers.ddpm_scheduler
• mmagic.models.editors.animatediff.animatediff
• mmagic.models.editors.animatediff.animatediff_utils
• mmagic.models.editors.animatediff.attention_3d
• mmagic.models.editors.animatediff.motion_module
• mmagic.models.editors.animatediff.resnet_3d
• mmagic.models.editors.animatediff.unet_3d
• mmagic.models.editors.animatediff.unet_block
• mmagic.models.editors.aotgan.aot_decoder
• mmagic.models.editors.aotgan.aot_encoder
• mmagic.models.editors.aotgan.aot_encoder_decoder
• mmagic.models.editors.aotgan.aot_inpaintor
• mmagic.models.editors.aotgan.aot_neck
• mmagic.models.editors.arcface.arcface_modules
• mmagic.models.editors.arcface.id_loss
• mmagic.models.editors.arcface.model_irse
• mmagic.models.editors.basicvsr.basicvsr
• mmagic.models.editors.basicvsr.basicvsr_net
• mmagic.models.editors.basicvsr_plusplus_net.basicvsr_plusplus_net
• mmagic.models.editors.biggan.biggan
• mmagic.models.editors.biggan.biggan_deep_discriminator
• mmagic.models.editors.biggan.biggan_deep_generator
• mmagic.models.editors.biggan.biggan_discriminator
• mmagic.models.editors.biggan.biggan_generator
• mmagic.models.editors.biggan.biggan_modules
• mmagic.models.editors.biggan.biggan_snmodule
• mmagic.models.editors.cain.cain
• mmagic.models.editors.cain.cain_net
• mmagic.models.editors.controlnet.controlnet
• mmagic.models.editors.controlnet.controlnet_utils
• mmagic.models.editors.cyclegan.cyclegan
• mmagic.models.editors.cyclegan.cyclegan_generator
• mmagic.models.editors.cyclegan.cyclegan_modules
• mmagic.models.editors.dcgan.dcgan
• mmagic.models.editors.dcgan.dcgan_discriminator
• mmagic.models.editors.dcgan.dcgan_generator
• mmagic.models.editors.ddpm.attention
• mmagic.models.editors.ddpm.denoising_unet
• mmagic.models.editors.ddpm.embeddings
• mmagic.models.editors.ddpm.res_blocks
• mmagic.models.editors.ddpm.unet_blocks
• mmagic.models.editors.deblurganv2.deblurganv2
• mmagic.models.editors.deblurganv2.deblurganv2_discriminator
• mmagic.models.editors.deblurganv2.deblurganv2_generator
• mmagic.models.editors.deblurganv2.deblurganv2_util
• mmagic.models.editors.deepfillv1.contextual_attention
• mmagic.models.editors.deepfillv1.contextual_attention_neck
• mmagic.models.editors.deepfillv1.deepfill_decoder
• mmagic.models.editors.deepfillv1.deepfill_disc
• mmagic.models.editors.deepfillv1.deepfill_encoder
• mmagic.models.editors.deepfillv1.deepfill_refiner
• mmagic.models.editors.deepfillv1.deepfillv1
• mmagic.models.editors.deepfillv2.two_stage_encoder_decoder
• mmagic.models.editors.dic.dic
• mmagic.models.editors.dic.dic_net
• mmagic.models.editors.dic.feedback_hour_glass
• mmagic.models.editors.dic.light_cnn
• mmagic.models.editors.dim.dim
• mmagic.models.editors.disco_diffusion.clip_wrapper
• mmagic.models.editors.disco_diffusion.disco
• mmagic.models.editors.disco_diffusion.guider
• mmagic.models.editors.disco_diffusion.secondary_model
• mmagic.models.editors.dreambooth.dreambooth
• mmagic.models.editors.duf.duf
• mmagic.models.editors.edsr.edsr_net
• mmagic.models.editors.edvr.edvr
• mmagic.models.editors.edvr.edvr_net
• mmagic.models.editors.eg3d.camera
• mmagic.models.editors.eg3d.dual_discriminator
• mmagic.models.editors.eg3d.eg3d
• mmagic.models.editors.eg3d.eg3d_generator
• mmagic.models.editors.eg3d.eg3d_modules
• mmagic.models.editors.eg3d.eg3d_utils
• mmagic.models.editors.eg3d.ray_sampler
• mmagic.models.editors.eg3d.renderer
• mmagic.models.editors.esrgan.esrgan
• mmagic.models.editors.esrgan.rrdb_net
• mmagic.models.editors.fastcomposer.fastcomposer
• mmagic.models.editors.fastcomposer.fastcomposer_util
• mmagic.models.editors.fba.fba_decoder
• mmagic.models.editors.fba.fba_encoder
• mmagic.models.editors.flavr.flavr
• mmagic.models.editors.flavr.flavr_net
• mmagic.models.editors.gca.gca
• mmagic.models.editors.gca.gca_module
• mmagic.models.editors.gca.resgca_dec
• mmagic.models.editors.gca.resgca_enc
• mmagic.models.editors.ggan.ggan
• mmagic.models.editors.glean.glean_styleganv2
• mmagic.models.editors.global_local.gl_decoder
• mmagic.models.editors.global_local.gl_dilation
• mmagic.models.editors.global_local.gl_disc
• mmagic.models.editors.global_local.gl_encoder
• mmagic.models.editors.global_local.gl_encoder_decoder
• mmagic.models.editors.global_local.gl_inpaintor
• mmagic.models.editors.guided_diffusion.adm
• mmagic.models.editors.guided_diffusion.classifier
• mmagic.models.editors.iconvsr.iconvsr_net
• mmagic.models.editors.indexnet.indexnet
• mmagic.models.editors.indexnet.indexnet_decoder
• mmagic.models.editors.indexnet.indexnet_encoder
• mmagic.models.editors.inst_colorization.color_utils
• mmagic.models.editors.inst_colorization.colorization_net
• mmagic.models.editors.inst_colorization.fusion_net
• mmagic.models.editors.inst_colorization.inst_colorization
• mmagic.models.editors.inst_colorization.weight_layer
• mmagic.models.editors.liif.liif
• mmagic.models.editors.liif.liif_net
• mmagic.models.editors.liif.mlp_refiner
• mmagic.models.editors.lsgan.lsgan
• mmagic.models.editors.lsgan.lsgan_discriminator
• mmagic.models.editors.lsgan.lsgan_generator
• mmagic.models.editors.mspie.mspie_stylegan2
• mmagic.models.editors.mspie.mspie_stylegan2_discriminator
• mmagic.models.editors.mspie.mspie_stylegan2_generator
• mmagic.models.editors.mspie.mspie_stylegan2_modules
• mmagic.models.editors.mspie.pe_singan
• mmagic.models.editors.mspie.pe_singan_generator
• mmagic.models.editors.mspie.positional_encoding
• mmagic.models.editors.nafnet.naf_avgpool2d
• mmagic.models.editors.nafnet.naf_layerNorm2d
• mmagic.models.editors.nafnet.nafbaseline_net
• mmagic.models.editors.nafnet.nafnet_net
• mmagic.models.editors.pconv.mask_conv_module
• mmagic.models.editors.pconv.partial_conv
• mmagic.models.editors.pconv.pconv_decoder
• mmagic.models.editors.pconv.pconv_encoder
• mmagic.models.editors.pconv.pconv_encoder_decoder
• mmagic.models.editors.pconv.pconv_inpaintor
• mmagic.models.editors.pggan.pggan
• mmagic.models.editors.pggan.pggan_discriminator
• mmagic.models.editors.pggan.pggan_generator
• mmagic.models.editors.pggan.pggan_modules
• mmagic.models.editors.pix2pix.pix2pix
• mmagic.models.editors.pix2pix.pix2pix_generator
• mmagic.models.editors.pix2pix.pix2pix_modules
• mmagic.models.editors.plain.plain_decoder
• mmagic.models.editors.plain.plain_refiner
• mmagic.models.editors.rdn.rdn_net
• mmagic.models.editors.real_basicvsr.real_basicvsr
• mmagic.models.editors.real_basicvsr.real_basicvsr_net
• mmagic.models.editors.real_esrgan.real_esrgan
• mmagic.models.editors.real_esrgan.unet_disc
• mmagic.models.editors.restormer.restormer_net
• mmagic.models.editors.sagan.sagan
• mmagic.models.editors.sagan.sagan_discriminator
• mmagic.models.editors.sagan.sagan_generator
• mmagic.models.editors.sagan.sagan_modules
• mmagic.models.editors.singan.singan
• mmagic.models.editors.singan.singan_discriminator
• mmagic.models.editors.singan.singan_generator
• mmagic.models.editors.singan.singan_modules
• mmagic.models.editors.srcnn.srcnn_net
• mmagic.models.editors.srgan.modified_vgg
• mmagic.models.editors.srgan.sr_resnet
• mmagic.models.editors.srgan.srgan
• mmagic.models.editors.stable_diffusion.clip_wrapper
• mmagic.models.editors.stable_diffusion.stable_diffusion
• mmagic.models.editors.stable_diffusion.stable_diffusion_inpaint
• mmagic.models.editors.stable_diffusion.vae
• mmagic.models.editors.stable_diffusion_xl.stable_diffusion_xl
• mmagic.models.editors.stylegan1.stylegan1
• mmagic.models.editors.stylegan1.stylegan1_discriminator
• mmagic.models.editors.stylegan1.stylegan1_generator
• mmagic.models.editors.stylegan1.stylegan1_modules
• mmagic.models.editors.stylegan1.stylegan_utils
• mmagic.models.editors.stylegan2.ada.augment
• mmagic.models.editors.stylegan2.ada.grid_sample_gradfix
• mmagic.models.editors.stylegan2.ada.misc
• mmagic.models.editors.stylegan2.ada.upfirdn2d
• mmagic.models.editors.stylegan2.stylegan2
• mmagic.models.editors.stylegan2.stylegan2_discriminator
• mmagic.models.editors.stylegan2.stylegan2_generator
• mmagic.models.editors.stylegan2.stylegan2_modules
• mmagic.models.editors.stylegan3.stylegan3
• mmagic.models.editors.stylegan3.stylegan3_generator
• mmagic.models.editors.stylegan3.stylegan3_modules
• mmagic.models.editors.stylegan3.stylegan3_utils
• mmagic.models.editors.swinir.swinir_modules
• mmagic.models.editors.swinir.swinir_net
• mmagic.models.editors.swinir.swinir_rstb
• mmagic.models.editors.swinir.swinir_utils
• mmagic.models.editors.tdan.tdan
• mmagic.models.editors.tdan.tdan_net
• mmagic.models.editors.textual_inversion.textual_inversion
• mmagic.models.editors.tof.tof_vfi_net
• mmagic.models.editors.tof.tof_vsr_net
• mmagic.models.editors.ttsr.lte
• mmagic.models.editors.ttsr.search_transformer
• mmagic.models.editors.ttsr.ttsr
• mmagic.models.editors.ttsr.ttsr_disc
• mmagic.models.editors.ttsr.ttsr_net
• mmagic.models.editors.vico.vico
• mmagic.models.editors.vico.vico_utils
• mmagic.models.editors.wgan_gp.wgan_discriminator
• mmagic.models.editors.wgan_gp.wgan_generator
• mmagic.models.editors.wgan_gp.wgan_gp
• mmagic.models.editors.wgan_gp.wgan_gp_module
• mmagic.models.losses.adv_loss
• mmagic.models.losses.clip_loss
• mmagic.models.losses.composition_loss
• mmagic.models.losses.face_id_loss
• mmagic.models.losses.feature_loss
• mmagic.models.losses.gan_loss
• mmagic.models.losses.gradient_loss
• mmagic.models.losses.loss_comps.clip_loss_comps
• mmagic.models.losses.loss_comps.disc_auxiliary_loss_comps
• mmagic.models.losses.loss_comps.face_id_loss_comps
• mmagic.models.losses.loss_comps.gan_loss_comps
• mmagic.models.losses.loss_comps.gen_auxiliary_loss_comps
• mmagic.models.losses.loss_wrapper
• mmagic.models.losses.perceptual_loss
• mmagic.models.losses.pixelwise_loss
• mmagic.models.utils.bbox_utils
• mmagic.models.utils.diffusion_utils
• mmagic.models.utils.flow_warp
• mmagic.models.utils.model_utils
• mmagic.models.utils.sampling_utils
• mmagic.models.utils.tensor_utils
• mmagic.models.utils.tome_utils
• mmagic.registry
• mmagic.structures.data_sample
• mmagic.utils.cli
• mmagic.utils.collect_env
• mmagic.utils.img_utils
• mmagic.utils.io_utils
• mmagic.utils.logger
• mmagic.utils.sampler
• mmagic.utils.setup_env
• mmagic.utils.trans_utils
• mmagic.utils.typing
• mmagic.version
• mmagic.visualization.concat_visualizer
• mmagic.visualization.vis_backend
• mmagic.visualization.visualizer
• paired_imgs_256x256_crop
• places
• realbasicvsr_c64b20_1x30x8_8xb1_lr5e_5_150k_reds
• realbasicvsr_wogan_c64b20_2x30x8_8xb2_lr1e_4_300k_reds
• sisr_x2_test_config
• sisr_x3_test_config
• sisr_x4_test_config
• stylegan2_c2_8xb4_800kiters_ffhq_256x256
• stylegan2_c2_8xb4_800kiters_lsun_cat_256x256
• stylegan2_c2_8xb4_800kiters_lsun_church_256x256
• stylegan2_c2_8xb4_800kiters_lsun_horse_256x256
• stylegan2_c2_8xb4_ffhq_1024x1024
• stylegan2_c2_8xb4_lsun_car_384x512
• stylegan2_c2_PL_R1_8xb4_apex_fp16_no_scaler_800kiters_ffhq_256x256
• stylegan3_r_ada_gamma33_8xb4_fp16_metfaces_1024x1024
• stylegan3_r_cvt_official_rgb_8xb4_ffhq_1024x1024
• stylegan3_r_cvt_official_rgb_8xb4_ffhqu_256x256
• stylegan3_r_cvt_official_rgb_8xb4x8_afhqv2_512x512
• stylegan3_t_ada_gamma66_8xb4_fp16_metfaces_1024x1024
• stylegan3_t_cvt_official_rgb_8xb4_afhqv2_512x512
• stylegan3_t_cvt_official_rgb_8xb4_ffhq_1024x1024
• stylegan3_t_cvt_official_rgb_8xb4_ffhqu_256x256
• stylegan3_t_gamma20_8xb4_fp16_noaug_ffhq_256x256
• stylegan3_t_gamma328_8xb4_fp16_noaug_ffhq_1024x1024
• tdan_test_config
• unconditional_imgs_128x128
• unconditional_imgs_64x64
• unconditional_imgs_flip_512x512
• unconditional_imgs_flip_lanczos_resize_256x256
• unpaired_imgs_256x256