from huggan.pytorch.pix2pix.modeling_pix2pix import GeneratorUNet model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')