Merge pull request #71 from kohya-ss/dev

Fix negative prompt not working when token>75
kohya-ss · Jan 15, 2023 · c2c1261 · c2c1261
2 parents 60e5793 + 48110bc
commit c2c1261
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gen_img_diffusers.py b/gen_img_diffusers.py
@@ -705,7 +705,7 @@ def __call__(
     if negative_scale is not None:
       _, real_uncond_embeddings, _ = get_weighted_text_embeddings(
           pipe=self,
-          prompt=[""]*batch_size,
+          prompt=prompt,                                   # こちらのトークン長に合わせてuncondを作るので75トークン超で必須
           uncond_prompt=[""]*batch_size,
           max_embeddings_multiples=max_embeddings_multiples,
           clip_skip=self.clip_skip,