move pipeline to cuda

app.py

@@ -317,6 +317,7 @@ class customUnClipPipeline(UnCLIPImageVariationPipeline):
 pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float32, trust_remote_code=True, 
                                             # accelerator='ort', device='cpu'
                                            )
+pipe.to('cuda')
 # pipe.enable_model_cpu_offload()