Skip to content

Commit f518f5f

Browse files
committed
make style
1 parent fe378b3 commit f518f5f

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,14 @@
1515
from ...models import AutoencoderKL, UNet2DConditionModel
1616
from ...models.lora import adjust_lora_scale_text_encoder
1717
from ...schedulers import KarrasDiffusionSchedulers
18-
from ...utils import USE_PEFT_BACKEND, is_torch_xla_available, BaseOutput, logging, scale_lora_layers, unscale_lora_layers
18+
from ...utils import (
19+
USE_PEFT_BACKEND,
20+
is_torch_xla_available,
21+
BaseOutput,
22+
logging,
23+
scale_lora_layers,
24+
unscale_lora_layers,
25+
)
1926
from ...utils.torch_utils import randn_tensor
2027
from ..pipeline_utils import DiffusionPipeline, StableDiffusionMixin
2128
from ..stable_diffusion import StableDiffusionSafetyChecker
@@ -288,7 +295,11 @@ def create_motion_field_and_warp_latents(motion_field_strength_x, motion_field_s
288295

289296

290297
class TextToVideoZeroPipeline(
291-
DiffusionPipeline, StableDiffusionMixin, TextualInversionLoaderMixin, StableDiffusionLoraLoaderMixin, FromSingleFileMixin
298+
DiffusionPipeline,
299+
StableDiffusionMixin,
300+
TextualInversionLoaderMixin,
301+
StableDiffusionLoraLoaderMixin,
302+
FromSingleFileMixin,
292303
):
293304
r"""
294305
Pipeline for zero-shot text-to-video generation using Stable Diffusion.
@@ -446,10 +457,10 @@ def backward_loop(
446457
if callback is not None and i % callback_steps == 0:
447458
step_idx = i // getattr(self.scheduler, "order", 1)
448459
callback(step_idx, t, latents)
449-
460+
450461
if XLA_AVAILABLE:
451462
xm.mark_step()
452-
463+
453464
return latents.clone().detach()
454465

455466
# Copied from diffusers.pipelines.stable_diffusion_k_diffusion.pipeline_stable_diffusion_k_diffusion.StableDiffusionKDiffusionPipeline.check_inputs

0 commit comments

Comments
 (0)