Menyu commited on
Commit
f7f9f9f
·
verified ·
1 Parent(s): 5627e44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -20,7 +20,6 @@ if torch.cuda.is_available():
20
  )
21
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
22
  pipe.scheduler.register_to_config(
23
- prediction_type="v_prediction",
24
  rescale_betas_zero_snr=True,
25
  )
26
  pipe.to("cuda")
 
20
  )
21
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
22
  pipe.scheduler.register_to_config(
 
23
  rescale_betas_zero_snr=True,
24
  )
25
  pipe.to("cuda")