LuChengTHU commited on
Commit
cdfe75a
β€’
1 Parent(s): a826c63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ scheduler = DPMSolverMultistepScheduler(
10
  beta_schedule="scaled_linear",
11
  num_train_timesteps=1000,
12
  trained_betas=None,
13
- predict_epsilon=True,
14
  thresholding=False,
15
  algorithm_type="dpmsolver++",
16
  solver_type="midpoint",
 
10
  beta_schedule="scaled_linear",
11
  num_train_timesteps=1000,
12
  trained_betas=None,
13
+ prediction_type="epsilon",
14
  thresholding=False,
15
  algorithm_type="dpmsolver++",
16
  solver_type="midpoint",