Update raw.py

raw.py

@@ -6,6 +6,7 @@ from diffusers import FluxControlNetModel, FluxControlNetPipeline, AutoencoderKL
 from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig
 from transformers import T5EncoderModel
 from transformers import BitsAndBytesConfig as TransformersBitsAndBytesConfig
+from peft import PeftModel, PeftConfig
 
 import gradio as gr
 huggingface_token = os.getenv("HUGGINFACE_TOKEN")
@@ -36,12 +37,14 @@ pipe = FluxControlNetPipeline.from_pretrained(
 )
 adapter_id = "alimama-creative/FLUX.1-Turbo-Alpha"
 adapter_id2 = "XLabs-AI/flux-RealismLora"
+adapter_id3 = "enhanceaiteam/Flux-uncensored-v2"
 
 pipe.to("cuda")
 pipe.load_lora_weights(adapter_id, adapter_name="turbo")
 pipe.load_lora_weights(adapter_id2, adapter_name="real")
-pipe.set_adapters(["turbo", "real"], adapter_weights=[0.9, 0.6])
-pipe.fuse_lora(adapter_names=["turbo", "real"], lora_scale=1.0)
+pipe.load_lora_weights(adapter_id3, weight_name="lora.safetensors", adapter_name="enhance")
+pipe.set_adapters(["turbo", "real", "enhance"], adapter_weights=[0.9, 0.66, 0.6])
+pipe.fuse_lora(adapter_names=["turbo", "real", "enhance"], lora_scale=1.0)
 pipe.unload_lora_weights()
 # pipe.enable_xformers_memory_efficient_attention()
 # save to the Hub