pass cuda device argument

app.py

@@ -23,12 +23,13 @@ conditioning_image_transforms = T.Compose(
     ]
 )
 
-cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True).to("cuda")
+cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True, device="0")
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
         "./models/wd-1-5-b2",
         controlnet=cnet,
         torch_dtype=torch.float16,
-        ).to("cuda")
+        device="0"
+        )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()