Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
•
2aeea2e
1
Parent(s):
aa5232a
Update modified_pipeline_semantic_stable_diffusion.py
Browse files
modified_pipeline_semantic_stable_diffusion.py
CHANGED
@@ -721,18 +721,18 @@ class SemanticStableDiffusionPipeline(DiffusionPipeline):
|
|
721 |
callback(i, t, latents)
|
722 |
|
723 |
|
724 |
-
|
725 |
-
image = self.decode_latents(latents)
|
726 |
|
727 |
-
# 9. Run safety checker
|
728 |
-
image, has_nsfw_concept = self.run_safety_checker(image, device, text_embeddings.dtype)
|
729 |
|
730 |
-
# 10. Convert to PIL
|
731 |
-
if output_type == "pil":
|
732 |
-
|
733 |
|
734 |
-
if not return_dict:
|
735 |
-
|
736 |
|
737 |
#return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
|
738 |
|
|
|
721 |
callback(i, t, latents)
|
722 |
|
723 |
|
724 |
+
# # 8. Post-processing
|
725 |
+
# image = self.decode_latents(latents)
|
726 |
|
727 |
+
# # 9. Run safety checker
|
728 |
+
# image, has_nsfw_concept = self.run_safety_checker(image, device, text_embeddings.dtype)
|
729 |
|
730 |
+
# # 10. Convert to PIL
|
731 |
+
# if output_type == "pil":
|
732 |
+
# image = self.numpy_to_pil(image)
|
733 |
|
734 |
+
# if not return_dict:
|
735 |
+
# return (image, has_nsfw_concept)
|
736 |
|
737 |
#return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
|
738 |
|