Linoy Tsaban commited on
Commit
996e7e0
1 Parent(s): 21466b8

Update pipeline_semantic_stable_diffusion_img2img_solver.py

Browse files
pipeline_semantic_stable_diffusion_img2img_solver.py CHANGED
@@ -1103,9 +1103,9 @@ class SemanticStableDiffusionImg2ImgPipeline_DPMSolver(DiffusionPipeline):
1103
  image = self.image_processor.postprocess(image, output_type=output_type, do_denormalize=do_denormalize)
1104
 
1105
  if not return_dict:
1106
- return (image, has_nsfw_concept)
1107
 
1108
- return SemanticStableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
1109
 
1110
  def encode_text(self, prompts):
1111
  text_inputs = self.tokenizer(
 
1103
  image = self.image_processor.postprocess(image, output_type=output_type, do_denormalize=do_denormalize)
1104
 
1105
  if not return_dict:
1106
+ return (image, has_nsfw_concept), attention_store
1107
 
1108
+ return SemanticStableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept), attention_store
1109
 
1110
  def encode_text(self, prompts):
1111
  text_inputs = self.tokenizer(