Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import numpy as np
 import spaces
 import torch
-from diffusers import FluxControlNetModel
+from diffusers import FluxControlNetModel, FluxPipeline
 from diffusers.pipelines import FluxControlNetPipeline
 from gradio_imageslider import ImageSlider
 from PIL import Image
@@ -45,8 +45,11 @@ controlnet = FluxControlNetModel.from_pretrained(
 pipe = FluxControlNetPipeline.from_pretrained(
     model_path, controlnet=controlnet, torch_dtype=torch.bfloat16
 )
+pipe.load_lora_weights(
+    hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), adapter_name="hyper-sd"
+)
+pipe.set_adapters(["hyper-sd"], adapter_weights=[0.125])
 pipe.to(device)
-
 MAX_SEED = 1000000
 MAX_PIXEL_BUDGET = 1024 * 1024