Spaces:
Build error
Build error
Update template/app_advanced.py
Browse files- template/app_advanced.py +1 -12
template/app_advanced.py
CHANGED
|
@@ -6,18 +6,7 @@ from PIL import Image
|
|
| 6 |
model_id = '$model_id'
|
| 7 |
prefix = '$prefix'
|
| 8 |
|
| 9 |
-
scheduler = DPMSolverMultistepScheduler(
|
| 10 |
-
beta_start=0.00085,
|
| 11 |
-
beta_end=0.012,
|
| 12 |
-
beta_schedule="scaled_linear",
|
| 13 |
-
num_train_timesteps=1000,
|
| 14 |
-
trained_betas=None,
|
| 15 |
-
predict_epsilon=True,
|
| 16 |
-
thresholding=False,
|
| 17 |
-
algorithm_type="dpmsolver++",
|
| 18 |
-
solver_type="midpoint",
|
| 19 |
-
lower_order_final=True,
|
| 20 |
-
)
|
| 21 |
|
| 22 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 23 |
model_id,
|
|
|
|
| 6 |
model_id = '$model_id'
|
| 7 |
prefix = '$prefix'
|
| 8 |
|
| 9 |
+
scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 12 |
model_id,
|