modify

app.py

@@ -95,13 +95,13 @@ def binary_mask_eval(args):
     # load sdxl model
     pipe = StableDiffusionXLPipeline.from_pretrained(
         "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.bfloat16
-    ).to(device)
+    )#.to(device)
 
     torch_dtype = torch.bfloat16 if args.mix_precision == "bf16" else torch.float32
     mask_pipe, hookers = create_pipeline(
         pipe,
         args.model,
-        device,
+        "cpu",
         torch_dtype,
         args.ckpt,
         binary=args.binary,
@@ -131,7 +131,7 @@ def binary_mask_eval(args):
     # reload the original model
     pipe = StableDiffusionXLPipeline.from_pretrained(
         "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.bfloat16
-    ).to(device)
+    )#.to(device)
 
     # get model param summary
     print(f"original model param: {get_model_param_summary(pipe.unet)['overall']}")