update

gradio_demo/app.py

@@ -131,7 +131,9 @@ def instantir_restore(
         map_location="cpu"
     )
     pipe.aggregator.load_state_dict(aggregator_state_dict, strict=True)
-    if not cpu_offload:
+    if cpu_offload:
+        pipe.aggregator.to(device="cpu")
+    else:
         pipe.aggregator.to(device=device, dtype=torch_dtype)
 
     print("******loaded")
@@ -147,8 +149,6 @@ def instantir_restore(
 #    pipe.enable_vae_tiling()
     if cpu_offload:
         pipe.enable_model_cpu_offload()
-        pipe.scheduler.to("cpu")
-        pipe.aggregator.to("cpu")
 #        pipe.enable_sequential_cpu_offload()
     print('done')