testing dtype change

app.py

@@ -314,12 +314,12 @@ class customUnClipPipeline(UnCLIPImageVariationPipeline):
 
 
 ### ADDITIONAL PIPELINE CODE FOR KARLO
-torch_device = 'cpu'
+torch_device = torch.device('cpu')
 pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float32, trust_remote_code=True, 
                                             # device=torch_device,
                                             # device_map='cpu'
                                            )
-pipe.to("cuda")
+pipe.to(torch.device("cuda"))
 # pipe.enable_model_cpu_offload()