Update app.py

app.py

@@ -19,6 +19,10 @@ if torch.cuda.is_available():
         add_watermarker=False
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    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: