Linoy Tsaban
commited on
Commit
·
e8414cc
1
Parent(s):
eec6d5e
Update modified_pipeline_semantic_stable_diffusion.py
Browse files
modified_pipeline_semantic_stable_diffusion.py
CHANGED
|
@@ -725,7 +725,7 @@ class SemanticStableDiffusionPipeline(DiffusionPipeline):
|
|
| 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":
|
|
|
|
| 725 |
image = self.decode_latents(latents)
|
| 726 |
|
| 727 |
# 9. Run safety checker
|
| 728 |
+
image, has_nsfw_concept = self.run_safety_checker(image, self.device, text_embeddings.dtype)
|
| 729 |
|
| 730 |
# 10. Convert to PIL
|
| 731 |
if output_type == "pil":
|