fix None prompt bug

adaface/adaface_wrapper.py

@@ -190,6 +190,9 @@ class AdaFaceWrapper(nn.Module):
             print(f"Updated {len(self.placeholder_token_ids)} tokens ({self.placeholder_tokens_str}) in the text encoder.")
 
     def update_prompt(self, prompt):
+        if prompt is None:
+            prompt = ""
+            
         # If the placeholder tokens are already in the prompt, then return the prompt as is.
         if self.placeholder_tokens_str in prompt:
             return prompt

app.py

@@ -109,6 +109,9 @@ def generate_image(image_container, uploaded_image_paths, init_img_file_paths, i
                    is_adaface_enabled, adaface_ckpt_path, adaface_id_cfg_scale, adaface_power_scale, 
                    adaface_anneal_steps, progress=gr.Progress(track_tqdm=True)):
 
+    if prompt is None:
+        prompt = ""
+
     prompt = prompt + " 8k uhd, high quality"
     if " shot" not in prompt:
         prompt = prompt + ", medium shot"
@@ -135,7 +138,7 @@ def generate_image(image_container, uploaded_image_paths, init_img_file_paths, i
             adaface.generate_adaface_embeddings(image_folder=None, image_paths=uploaded_image_paths,
                                                 out_id_embs_scale=adaface_id_cfg_scale, update_text_encoder=True)
             # adaface_prompt_embeds: [1, 77, 768].
-            adaface_prompt_embeds, _ = adaface.encode_prompt(prompt)
+            adaface_prompt_embeds, _ = adaface.encode_prompt(prompt, verbose=True)
 
     # init_img_file_paths is a list of image paths. If not chose, init_img_file_paths is None.
     if init_img_file_paths is not None: