Update app.py

app.py

@@ -150,6 +150,7 @@ def use_output_as_input(output_image):
 
 # Initialize Qwen Image Edit pipeline
 # Scheduler configuration for Lightning
+dtype = torch.bfloat16
 scheduler_config = {
     "base_image_seq_len": 256,
     "base_shift": math.log(3),