m

entklei.py

@@ -233,24 +233,12 @@ def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positi
 
     dwpose_pil = dwpose(person_cropped_pil, 512)
 
-
     expanded_mask_image_width, expanded_mask_image_height = expanded_mask_image.size
 
     dwpose_pil_resized = dwpose_pil.resize(
         (int(expanded_mask_image_width), int(expanded_mask_image_height)))
 
-    pipe = base_pipe
-
-    if with_big_tits or with_small_tits:
-        pipe = pipe_with_tit_slider
-
-    # if with_small_tits:
-    #     pipe = pipe_with_small_tits
-
-    # if with_big_tits:
-    #     pipe = pipe_with_big_tits
-
-    end_result_images = pipe(
+    end_result_images = pipe_with_tit_slider(
         positive_prompt,
         negative_prompt=negative_prompt,
         num_inference_steps=steps,
@@ -262,7 +250,7 @@ def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positi
         num_images_per_prompt=2,
         height=round(new_size["height"]),
         width=round(new_size["width"]),
-        cross_attention_kwargs={"scale": -0.8} if with_small_tits else {"scale": 0.7} if with_big_tits else {}
+        cross_attention_kwargs={"scale": -0.8} if with_small_tits else {"scale": 0.7} if with_big_tits else { "scale": 0 }
     ).images
 
     # Function to create a mask for blurring edges