linoyts HF staff commited on
Commit
c9c54e7
1 Parent(s): 1e4841f

Update clip_slider_pipeline.py

Browse files
Files changed (1) hide show
  1. clip_slider_pipeline.py +1 -1
clip_slider_pipeline.py CHANGED
@@ -457,7 +457,7 @@ class CLIPSliderFlux(CLIPSlider):
457
  images = self.pipe(prompt_embeds=prompt_embeds, pooled_prompt_embeds=pooled_prompt_embeds,
458
  **pipeline_kwargs).images
459
 
460
- return images
461
 
462
  def spectrum(self,
463
  prompt="a photo of a house",
 
457
  images = self.pipe(prompt_embeds=prompt_embeds, pooled_prompt_embeds=pooled_prompt_embeds,
458
  **pipeline_kwargs).images
459
 
460
+ return images[0]
461
 
462
  def spectrum(self,
463
  prompt="a photo of a house",