update

gradio_demo/app.py

@@ -125,7 +125,7 @@ def instantir_restore(
     print("Loading checkpoint...")
     aggregator_state_dict = torch.load(
         f"{instantir_path}/aggregator.pt",
-        map_location="cpu"
+#        map_location="cpu"
     )
     pipe.aggregator.load_state_dict(aggregator_state_dict, strict=True)
     pipe.aggregator.to(device=device, dtype=torch_dtype)
@@ -140,10 +140,10 @@ def instantir_restore(
             pipe.unet.set_adapter('previewer')
 
     print('optimizing')
-    pipe.enable_vae_tiling()
+#    pipe.enable_vae_tiling()
     if cpu_offload:
-#        pipe.enable_model_cpu_offload()
-        pipe.enable_sequential_cpu_offload()
+        pipe.enable_model_cpu_offload()
+#        pipe.enable_sequential_cpu_offload()
     print('done')