m

entklei.py

@@ -135,13 +135,13 @@ pipe_with_tit_slider = _.clone_deep(base_pipe)
 
 pipe_with_tit_slider.load_lora_weights(os.path.join("/repository" if is_production else ".", "models", "breastsizeslideroffset.safetensors"), weight_name="breastsizeslideroffset.safetensors", adapter_name="breastsizeslideroffset")
 
-pipe_with_small_tits = _.clone_deep(pipe_with_tit_slider)
-pipe_with_small_tits.set_adapters("breastsizeslideroffset", adapter_weights=[-0.8])
+# pipe_with_small_tits = _.clone_deep(pipe_with_tit_slider)
+# pipe_with_small_tits.set_adapters("breastsizeslideroffset", adapter_weights=[-0.8])
 
-pipe_with_medium_tits = _.clone_deep(base_pipe)
+# pipe_with_medium_tits = _.clone_deep(base_pipe)
 
-pipe_with_big_tits = _.clone_deep(pipe_with_tit_slider)
-pipe_with_big_tits.set_adapters("breastsizeslideroffset", adapter_weights=[0.7])
+# pipe_with_big_tits = _.clone_deep(pipe_with_tit_slider)
+# pipe_with_big_tits.set_adapters("breastsizeslideroffset", adapter_weights=[0.7])
 
 def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positive_prompt="nude girl, pussy, tits", negative_prompt="ugly", steps=20, cfg_scale=7, get_mask_function=None, with_small_tits=False, with_big_tits=False):
     try:
@@ -241,11 +241,14 @@ def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positi
 
     pipe = base_pipe
 
-    if with_small_tits:
-        pipe = pipe_with_small_tits
+    if with_big_tits or with_small_tits:
+        pipe = pipe_with_tit_slider
 
-    if with_big_tits:
-        pipe = pipe_with_big_tits
+    # if with_small_tits:
+    #     pipe = pipe_with_small_tits
+
+    # if with_big_tits:
+    #     pipe = pipe_with_big_tits
 
     end_result_images = pipe(
         positive_prompt,
@@ -258,7 +261,8 @@ def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positi
         control_image=dwpose_pil_resized,
         num_images_per_prompt=2,
         height=round(new_size["height"]),
-        width=round(new_size["width"])
+        width=round(new_size["width"]),
+        cross_attention_kwargs={"scale": -0.8} if with_small_tits else {"scale": 0.7} if with_big_tits else {}
     ).images
 
     # Function to create a mask for blurring edges

handler.py

@@ -26,7 +26,7 @@ if is_production:
     sys.path.insert(0, '/repository')
 
 if install_stuff:
-    result = subprocess.run(['pip', 'install', "-u", 'peft'], check=True)
+    # result = subprocess.run(['pip', 'install', "-u", 'peft'], check=True)
     result = subprocess.run(['pip', 'install', '-e', 'GroundingDINO'], check=True)
     print(f'pip install GroundingDINO = {result}')