Update app.py

app.py

@@ -233,6 +233,10 @@ if torch.cuda.is_available():
         use_safetensors=True,
         add_watermarker=False
     )
+    pipe.scheduler.register_to_config(
+        prediction_type="v_prediction",
+        rescale_betas_zero_snr=True,
+    )
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: