Update app.py

app.py

@@ -19,8 +19,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev",
                                          custom_pipeline="pipeline_flux_rf_inversion",
                                           torch_dtype=torch.bfloat16)
-pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
-pipe.fuse_lora(lora_scale=0.125)
+# pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
+# pipe.fuse_lora(lora_scale=0.125)
 pipe.to(DEVICE)
 
 examples = [[Image.open("cat.jpg"), "a tiger"]]
@@ -59,9 +59,9 @@ def invert_and_edit(image,
     
     
     output = pipe(prompt, 
-    inverted_latents=inverted_latents,
-    image_latents=image_latents,
-    latent_image_ids=latent_image_ids,
+    inverted_latents=inverted_latents.to(DEVICE),
+    image_latents=image_latents.to(DEVICE),
+    latent_image_ids=latent_image_ids.to(DEVICE),
     start_timestep=start_timestep, 
     stop_timestep=stop_timestep,
     num_inference_steps=num_inversion_steps,