update

gradio_demo/app.py

@@ -126,9 +126,9 @@ def instantir_restore(
     print("Loading checkpoint...")
     aggregator_state_dict = torch.load(
         f"{instantir_path}/aggregator.pt",
-        map_location = device
+#        map_location = device
 #        map_location = device if not cpu_offload else "cpu"
-#        map_location="cpu"
+        map_location="cpu"
     )
     pipe.aggregator.load_state_dict(aggregator_state_dict, strict=True)
     if not cpu_offload:
@@ -167,9 +167,9 @@ def instantir_restore(
         lq = [resize_img(lq.convert("RGB"), size=None)]
    
     if cpu_offload:
-        generator = torch.Generator(device=device).manual_seed(seed)
-    else:
         generator = torch.Generator().manual_seed(seed)
+    else:
+        generator = torch.Generator(device=device).manual_seed(seed)
     timesteps = [
         i * (1000//steps) + pipe.scheduler.config.steps_offset for i in range(0, steps)
     ]