HuiZhang commited on
Commit
e8676af
·
verified ·
1 Parent(s): 3beddd6

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()