safety guidance in sld

joycenerd · Nov 26, 2024 · 9615e7d · 9615e7d
1 parent f625627
commit 9615e7d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/model/p4dk/modified_sld_pipeline.py b/model/p4dk/modified_sld_pipeline.py
@@ -144,7 +144,7 @@ def _expand_safe_text_embeddings(self, text_embeddings, num_images_per_prompt):
         return text_embeddings
 
 
-    def _new_encode_prompt(self, prompt, num_images_per_prompt, do_classifier_free_guidance, negative_prompt, enable_safety_guidance, prompt_ids=None, prompt_embeddings=None):
+    def _new_encode_prompt(self, prompt, num_images_per_prompt, do_classifier_free_guidance, negative_prompt, enable_safety_guidance=True, prompt_ids=None, prompt_embeddings=None):
         r"""
         Encodes the prompt into text encoder hidden states.
         Args:

diff --git a/model/p4dn/modified_sld_pipeline.py b/model/p4dn/modified_sld_pipeline.py
@@ -155,7 +155,7 @@ def _expand_safe_text_embeddings(self, text_embeddings, num_images_per_prompt):
         return text_embeddings
 
 
-    def _new_encode_prompt(self, prompt, num_images_per_prompt, do_classifier_free_guidance, negative_prompt, enable_safety_guidance, prompt_ids=None, prompt_embeddings=None):
+    def _new_encode_prompt(self, prompt, num_images_per_prompt, do_classifier_free_guidance, negative_prompt, enable_safety_guidance=True, prompt_ids=None, prompt_embeddings=None):
         r"""
         Encodes the prompt into text encoder hidden states.
         Args: