Spaces:
Running
on
Zero
Running
on
Zero
HuiZhang
commited on
Update src/pipeline/pipeline_CreatiLayout.py
Browse files
src/pipeline/pipeline_CreatiLayout.py
CHANGED
@@ -906,7 +906,6 @@ class CreatiLayoutSD3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingleF
|
|
906 |
masks[:n_objs] = 1
|
907 |
repeat_batch = batch_size * num_images_per_prompt
|
908 |
boxes = boxes.unsqueeze(0).expand(repeat_batch, -1, -1).clone()
|
909 |
-
boxex = boxes.to(device=device, dtype=latents.dtype)
|
910 |
text_embeddings = text_embeddings.unsqueeze(0).expand(repeat_batch, -1, -1).clone()
|
911 |
text_embeddings = text_embeddings.to(device=device, dtype=latents.dtype)
|
912 |
masks = masks.unsqueeze(0).expand(repeat_batch, -1).clone()
|
|
|
906 |
masks[:n_objs] = 1
|
907 |
repeat_batch = batch_size * num_images_per_prompt
|
908 |
boxes = boxes.unsqueeze(0).expand(repeat_batch, -1, -1).clone()
|
|
|
909 |
text_embeddings = text_embeddings.unsqueeze(0).expand(repeat_batch, -1, -1).clone()
|
910 |
text_embeddings = text_embeddings.to(device=device, dtype=latents.dtype)
|
911 |
masks = masks.unsqueeze(0).expand(repeat_batch, -1).clone()
|