Update pipeline.py
Browse files- pipeline.py +0 -1
pipeline.py
CHANGED
|
@@ -887,7 +887,6 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
| 887 |
if torch.backends.mps.is_available():
|
| 888 |
# some platforms (eg. apple mps) misbehave due to a pytorch bug: https://github.com/pytorch/pytorch/pull/99272
|
| 889 |
latents = latents.to(latents_dtype)
|
| 890 |
-
)
|
| 891 |
|
| 892 |
# call the callback, if provided
|
| 893 |
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
|
|
|
|
| 887 |
if torch.backends.mps.is_available():
|
| 888 |
# some platforms (eg. apple mps) misbehave due to a pytorch bug: https://github.com/pytorch/pytorch/pull/99272
|
| 889 |
latents = latents.to(latents_dtype)
|
|
|
|
| 890 |
|
| 891 |
# call the callback, if provided
|
| 892 |
if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
|