not-lain commited on
Commit
e3a2cfa
1 Parent(s): 3a4f539

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ scheduler = DPMSolverMultistepScheduler(
35
  beta_schedule="scaled_linear",
36
  num_train_timesteps=1000,
37
  trained_betas=None,
38
- predict_epsilon=True,
39
  thresholding=False,
40
  algorithm_type="dpmsolver++",
41
  solver_type="midpoint",
 
35
  beta_schedule="scaled_linear",
36
  num_train_timesteps=1000,
37
  trained_betas=None,
38
+ prediction_type="epsilon",
39
  thresholding=False,
40
  algorithm_type="dpmsolver++",
41
  solver_type="midpoint",