text: from .cnn_discriminator import CNN
  from .cnn_generator import CNNGenerator
  from .stylegan2_discriminator import StyleGanDiscriminator
  from .vitgan_discriminator import ViT
  from .vitgan_generator import GeneratorViT
  from .diffaugment import DiffAugment
  from .utils import spectral_norm
  
  __all__ = [CNN, CNNGenerator, StyleGanDiscriminator, ViT, GeneratorViT, DiffAugment, spectral_norm]
  