Update app.py
Browse files
app.py
CHANGED
|
@@ -116,7 +116,7 @@ def load_and_prepare_model(model_id):
|
|
| 116 |
add_watermarker=False,
|
| 117 |
use_safetensors=True,
|
| 118 |
vae=vae,
|
| 119 |
-
|
| 120 |
)
|
| 121 |
sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start =0.00085,beta_end =0.012,steps_offset =1,)
|
| 122 |
pipe.scheduler=sched
|
|
|
|
| 116 |
add_watermarker=False,
|
| 117 |
use_safetensors=True,
|
| 118 |
vae=vae,
|
| 119 |
+
# scheduler=sched
|
| 120 |
)
|
| 121 |
sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start =0.00085,beta_end =0.012,steps_offset =1,)
|
| 122 |
pipe.scheduler=sched
|