3 Nb@s@ddlZddlmZddlmZmZmZmZddZddZ dS)N)FFCResNetGenerator)GlobalGeneratorMultiDilatedGlobalGeneratorNLayerDiscriminatorMultidilatedNLayerDiscriminatorcKsXtjd||dkr"tf|S|dkr4tf|S|dkrFtf|Std|dS)NzMake generator Zpix2pixhd_multidilatedZpix2pixhd_globalZ ffc_resnetzUnknown generator kind )logginginforrr ValueError)configkindkwargsr I/home/ec2-user/SageMaker/lama/saicinpainting/training/modules/__init__.pymake_generators   rcKsFtjd||dkr"tf|S|dkr4tf|Std|dS)NzMake discriminator Zpix2pixhd_nlayer_multidilatedZpix2pixhd_nlayerzUnknown discriminator kind )rrrrr )r r r r rmake_discriminators   r) rZ#saicinpainting.training.modules.ffcrZ)saicinpainting.training.modules.pix2pixhdrrrrrrr r r rs