Use GPU always

app.py

@@ -60,9 +60,9 @@ with gr.Blocks(css="style.css") as demo:
         scheduler_class = MyEulerAncestralDiscreteScheduler
         
         print('\n################## 1')
-        pipe_inversion = SDXLDDIMPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to('cpu')
+        pipe_inversion = SDXLDDIMPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)#.to('cpu')
         print('\n################## 2')
-        pipe_inference = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to('cpu')
+        pipe_inference = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)#.to('cpu')
         print('\n################## 3')
         pipe_inference.scheduler = scheduler_class.from_config(pipe_inference.scheduler.config)
         pipe_inversion.scheduler = scheduler_class.from_config(pipe_inversion.scheduler.config)
@@ -75,7 +75,7 @@ with gr.Blocks(css="style.css") as demo:
                            edit_guidance_scale=edit_guidance_scale,
                            inversion_max_step=inversion_max_step)
         image_editor = ImageEditorDemo(pipe_inversion, pipe_inference, input_image,
-                               description_prompt, config, 'cpu',
+                               description_prompt, config, device,
                                [rnri_iterations, rnri_alpha, rnri_lr])
         print('\n################## 5')
         return image_editor, "Input has set!"
@@ -84,10 +84,10 @@ with gr.Blocks(css="style.css") as demo:
     def edit(editor, target_prompt):
         if editor is None:
             raise gr.Error("Set inputs before editing.")
-        if device == "cuda":
-            image = editor.to(device).edit(target_prompt)
-        else:
-            image = editor.edit(target_prompt)
+       # if device == "cuda":
+     #       image = editor.to(device).edit(target_prompt)
+      #  else:
+        image = editor.edit(target_prompt)
         return image
 
     gr.Markdown(f"""running on {power_device}""")