diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -176,7 +176,7 @@ def create_pipeline(model, scheduler, dtype=torch.bfloat16): | |||
176 | ) | 176 | ) |
177 | else: | 177 | else: |
178 | scheduler = EulerAScheduler( | 178 | scheduler = EulerAScheduler( |
179 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False | 179 | beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear" |
180 | ) | 180 | ) |
181 | 181 | ||
182 | pipeline = VlpnStableDiffusion( | 182 | pipeline = VlpnStableDiffusion( |