Update app.py
Browse files
app.py
CHANGED
|
@@ -159,8 +159,8 @@ def infer(
|
|
| 159 |
initial_latents = pipe.prepare_latents(
|
| 160 |
batch_size=1,
|
| 161 |
num_channels_latents=pipe.transformer.in_channels,
|
| 162 |
-
height=pipe.transformer.sample_size,
|
| 163 |
-
width=pipe.transformer.sample_size,
|
| 164 |
dtype=pipe.transformer.dtype,
|
| 165 |
device=pipe.device,
|
| 166 |
generator=generator,
|
|
@@ -206,8 +206,8 @@ def infer(
|
|
| 206 |
initial_latents = pipe.prepare_latents(
|
| 207 |
batch_size=1,
|
| 208 |
num_channels_latents=pipe.transformer.in_channels,
|
| 209 |
-
height=pipe.transformer.sample_size[0],
|
| 210 |
-
width=pipe.transformer.sample_size[1],
|
| 211 |
dtype=pipe.transformer.dtype,
|
| 212 |
device=pipe.device,
|
| 213 |
generator=generator,
|
|
|
|
| 159 |
initial_latents = pipe.prepare_latents(
|
| 160 |
batch_size=1,
|
| 161 |
num_channels_latents=pipe.transformer.in_channels,
|
| 162 |
+
height=pipe.transformer.config.sample_size[0],
|
| 163 |
+
width=pipe.transformer.config.sample_size[1],
|
| 164 |
dtype=pipe.transformer.dtype,
|
| 165 |
device=pipe.device,
|
| 166 |
generator=generator,
|
|
|
|
| 206 |
initial_latents = pipe.prepare_latents(
|
| 207 |
batch_size=1,
|
| 208 |
num_channels_latents=pipe.transformer.in_channels,
|
| 209 |
+
height=pipe.transformer.config.sample_size[0],
|
| 210 |
+
width=pipe.transformer.config.sample_size[1],
|
| 211 |
dtype=pipe.transformer.dtype,
|
| 212 |
device=pipe.device,
|
| 213 |
generator=generator,
|