Spaces:

wl-zhao
/

unipc_sdm

Runtime error

zwl commited on Feb 17, 2023

Commit

0e16277

1 Parent(s): e9cfc77

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,10 +9,11 @@ scheduler = UniPCMultistepScheduler(
     beta_end =0.012,
     solver_order=2,
     prediction_type="epsilon",
     thresholding=False,
     solver_type='bh2',
     lower_order_final=True,
-    disable_corrector=[],
 )
 class Model:

     beta_end =0.012,
     solver_order=2,
     prediction_type="epsilon",
+    predict_x0=True,
     thresholding=False,
     solver_type='bh2',
     lower_order_final=True,
+    disable_corrector=[0],
 )
 class Model: