Hugging Face's logo

Spaces:
wl-zhao
/
unipc_sdm
Runtime error

App Files Community
1
zwl commited on
Commit
4364727
·
1 Parent(s): 0e16277

scaled_linear

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,7 +6,8 @@ import os
6
 
7
  scheduler = UniPCMultistepScheduler(
8
  beta_start=0.00085,
9
- beta_end =0.012,
 
10
  solver_order=2,
11
  prediction_type="epsilon",
12
  predict_x0=True,
 
6
 
7
  scheduler = UniPCMultistepScheduler(
8
  beta_start=0.00085,
9
+ beta_end=0.012,
10
+ beta_schedule="scaled_linear",
11
  solver_order=2,
12
  prediction_type="epsilon",
13
  predict_x0=True,