Update Time_TravelRephotography/projector.py

Time_TravelRephotography/projector.py

@@ -75,9 +75,9 @@ def create_generator(file_name: str, path:str,args: Namespace, device: torch.dev
                            use_auth_token=TOKEN)
     with open(path, 'rb') as f:
         generator = Generator(args.generator_size, 512, 8)
-        generator.load_state_dict(torch.load(f)['g_ema'], strict=False)
+        generator.load_state_dict(torch.load(f)['G_ema'], strict=False)
     generator.eval()
-    generator = generator.to(device)
+    generator.to(device)
     return generator