linoyts HF staff commited on
Commit
226559a
1 Parent(s): cf3bc31

Update clip_slider_pipeline.py

Browse files
Files changed (1) hide show
  1. clip_slider_pipeline.py +1 -1
clip_slider_pipeline.py CHANGED
@@ -481,6 +481,6 @@ class T5SliderFlux(CLIPSlider):
481
 
482
  torch.manual_seed(seed)
483
  images = self.pipe(prompt_embeds=prompt_embeds, pooled_prompt_embeds=pooled_prompt_embeds,
484
- **pipeline_kwargs).images
485
 
486
  return images
 
481
 
482
  torch.manual_seed(seed)
483
  images = self.pipe(prompt_embeds=prompt_embeds, pooled_prompt_embeds=pooled_prompt_embeds,
484
+ **pipeline_kwargs).images[0]
485
 
486
  return images