a

handler.py

@@ -64,8 +64,6 @@ class EndpointHandler():
 
         replicate_api_key = data.get("replicate_api_key", "")
 
-        print("got repli", replicate_api_key)
-
         with_small_tits = data.get("with_small_tits", False)
 
         with_big_tits = data.get("with_big_tits", False)

main.py

@@ -187,14 +187,14 @@ pipe_with_tit_slider.load_lora_weights(os.path.join(os.environ.get(
         'path', "."), "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.5])
+pipe_with_small_tits.set_adapters("breastsizeslideroffset", adapter_weights=[-0.8])
 
 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=[1])
+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, loras=[], lora_weights=[], replicate_api_key="", with_small_tits=False, with_big_tits=False):
+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, replicate_api_key="", with_small_tits=False, with_big_tits=False):
     try:
         exif_data = original_pil._getexif()
         orientation_tag = 274  # The Exif tag for orientation
@@ -289,13 +289,13 @@ def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positi
     dwpose_pil_resized = dwpose_pil.resize(
         (int(expanded_mask_image_width), int(expanded_mask_image_height)))
 
-    pipe = _.clone_deep(base_pipe)
+    pipe = base_pipe
 
     if with_small_tits:
-        pipe = _.clone_deep(pipe_with_small_tits)
+        pipe = pipe_with_small_tits
 
     if with_big_tits:
-        pipe = _.clone_deep(pipe_with_big_tits)
+        pipe = pipe_with_big_tits
 
     end_result_images = pipe(
         positive_prompt,
@@ -311,10 +311,6 @@ def get_nude(original_pil, original_max_size=2000, generate_max_size=768, positi
         width=round(new_size["width"])
     ).images
 
-    pipe.disable_lora()
-
-    del pipe
-
     # Function to create a mask for blurring edges
     def create_blurred_edge_mask(image, blur_radius):
         mask = Image.new("L", image.size, 0)