Spaces:

LuChengTHU
/

dpmsolver_sdm

Runtime error

LuChengTHU commited on Apr 11, 2023

Commit

cdfe75a

1 Parent(s): a826c63

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ scheduler = DPMSolverMultistepScheduler(
     beta_schedule="scaled_linear",
     num_train_timesteps=1000,
     trained_betas=None,
-    predict_epsilon=True,
     thresholding=False,
     algorithm_type="dpmsolver++",
     solver_type="midpoint",

     beta_schedule="scaled_linear",
     num_train_timesteps=1000,
     trained_betas=None,
+    prediction_type="epsilon",
     thresholding=False,
     algorithm_type="dpmsolver++",
     solver_type="midpoint",