Update inference.py

inference.py

@@ -136,7 +136,7 @@ class Inference(object):
         """Restore the trained generator and discriminator."""
         print('Loading the model...')
         G_path = os.path.join(model_directory, '{}-G.ckpt'.format(submodel))
-        self.G.load_state_dict(torch.load(G_path, map_location=lambda storage, loc: storage))
+        self.G.load_state_dict(torch.load(G_path, map_location=lambda storage, loc: storage, weights_only=False))
 
     def inference(self):
         # Load the trained generator.