push

app.py

@@ -62,13 +62,16 @@ pipe.load_lora_weights(
 pipe.load_lora_weights("xey/sldr_flux_nsfw_v2-studio", weight_name="sldr_flux_nsfw_v2-studio.safetensors", adapter_name="sldr")
 pipe.set_adapters(["hyper-sd", "hina", "sf5", "sldr"], adapter_weights=[0.25, 0.5, 0.25, 0.25])
 pipe.fuse_lora(lora_scale=0.8)
-pipe.unload_lora_weights()
 
 pipe.transformer.to(torch.bfloat16)
 pipe.controlnet.to(torch.bfloat16)
+pipe.push_to_hub("FLUX.1-Inpainting-8step_uncensored", private=True, token=HF_TOKEN)
+
+# pipe.unload_lora_weights()
+
 # pipe.to("cuda")
 pipe.enable_model_cpu_offload()
-
+print(pipe.hf_device_map)
 
 def create_mask_from_editor(editor_value):
     """
@@ -158,7 +161,7 @@ def inpaint_image(image, prompt, subject, editor_value):
 
     subject_name=subject
     target_text_prompt=prompt
-    prompt_final=f'A two side-by-side image of same {subject_name}. LEFT: a photo of the {subject_name}; RIGHT: a photo of the {subject_name} {target_text_prompt}.'
+    prompt_final=f'A two side-by-side image of {subject_name}. LEFT: a photo of {subject_name}; RIGHT: a photo of {subject_name} {target_text_prompt}.'
 
     # Convert attention mask to PIL image format
     # Take first head's mask after prompt tokens (shape is now H*W x H*W)
@@ -179,7 +182,7 @@ def inpaint_image(image, prompt, subject, editor_value):
         control_mask=mask,
         num_inference_steps=8,
         generator=generator,
-        controlnet_conditioning_scale=0.9,
+        controlnet_conditioning_scale=0.6,
         guidance_scale=3.5,
         negative_prompt="",
         true_guidance_scale=1.0,
@@ -195,8 +198,10 @@ with gr.Blocks() as iface:
     with gr.Row():
         with gr.Column():
             with gr.Row():
-                input_image = gr.Image(type="filepath", label="Upload Image")
+                with gr.Accordion():
+                    input_image = gr.Image(type="filepath", label="Upload Image")
             with gr.Row():
+                prompt_preview = gr.Textbox(value="A two side-by-side image of 'subject_name'. LEFT: a photo of 'subject_name'; RIGHT: a photo of 'subject_name' 'target_text_prompt'", interactive=False)
                 subject = gr.Textbox(lines=1, placeholder="Enter your subject", label="Subject")
                 prompt = gr.Textbox(lines=2, placeholder="Enter your prompt here (e.g., 'wearing a christmas hat, in a busy street')", label="Prompt")
         with gr.Column():