StableDiffusion-3.5-Large

Running on Zero

ford442 commited on Dec 10, 2024

Commit

b809e76

verified ·

1 Parent(s): f238e99

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -159,8 +159,8 @@ def infer(
         initial_latents = pipe.prepare_latents(
             batch_size=1,
             num_channels_latents=pipe.transformer.in_channels,
-            height=pipe.transformer.sample_size,
-            width=pipe.transformer.sample_size,
             dtype=pipe.transformer.dtype,
             device=pipe.device,
             generator=generator,
@@ -206,8 +206,8 @@ def infer(
         initial_latents = pipe.prepare_latents(
             batch_size=1,
             num_channels_latents=pipe.transformer.in_channels,
-            height=pipe.transformer.sample_size[0],
-            width=pipe.transformer.sample_size[1],
             dtype=pipe.transformer.dtype,
             device=pipe.device,
             generator=generator,

         initial_latents = pipe.prepare_latents(
             batch_size=1,
             num_channels_latents=pipe.transformer.in_channels,
+            height=pipe.transformer.config.sample_size[0],
+            width=pipe.transformer.config.sample_size[1],
             dtype=pipe.transformer.dtype,
             device=pipe.device,
             generator=generator,
         initial_latents = pipe.prepare_latents(
             batch_size=1,
             num_channels_latents=pipe.transformer.in_channels,
+            height=pipe.transformer.config.sample_size[0],
+            width=pipe.transformer.config.sample_size[1],
             dtype=pipe.transformer.dtype,
             device=pipe.device,
             generator=generator,