              print ("Generator models loaded.")
          except:
              print ("Generator weights files not found.")
              pass
          return netGA, netGB
  
      def build_discriminator(self):
          def conv_block_d(input_tensor, f, use_instance_norm=True):
 
</s> remove         self.netGA.save_weights(str(self.model_dir / netGAH5))
        self.netGB.save_weights(str(self.model_dir / netGBH5))
 </s> add         if self.gpus > 1:
            self.netGA_sm.save_weights(str(self.model_dir / netGAH5))
            self.netGB_sm.save_weights(str(self.model_dir / netGBH5))
        else:
            self.netGA.save_weights(str(self.model_dir / netGAH5))
            self.netGB.save_weights(str(self.model_dir / netGBH5))
 </s> remove         print ("Models saved.") </s> add  </s> remove     def __init__(self, model_dir):
 </s> add     def __init__(self, model_dir, gpus):


        if self.gpus > 1:
            netGA = multi_gpu_model( self.netGA_sm , self.gpus)
            netGB = multi_gpu_model( self.netGB_sm , self.gpus)


              print ("Generator models loaded.")
          except:
              print ("Generator weights files not found.")
              pass

        if self.gpus > 1:
            netGA = multi_gpu_model( self.netGA_sm , self.gpus)
            netGB = multi_gpu_model( self.netGB_sm , self.gpus)

          return netGA, netGB
  
      def build_discriminator(self):
          def conv_block_d(input_tensor, f, use_instance_norm=True):
keep keep keep add keep keep keep keep
 Fix broken multigpu GAN loading. (#280)

* Fix broken multigpu GAN loading.

Fix for loading into the multi GPU model when it needs to load/save the original model.

* Move reference change.

Move reference change for single model so it is defined before load

* Add mutliGPU support to GAN128

Added support for mutligpu to GAN128 
plugins/Model_GAN/Model.py