diffusion main

app.py

@@ -1,13 +1,13 @@
 import torch
 import gradio as gr
 from diffusers.pipelines.flux.pipeline_flux import FluxPipeline
-from diffusers.models.controlnet import ControlNetModel
+from diffusers.models.controlnet_flux import FluxControlNetModel
 from controlnet_aux import CannyDetector
 
 base_model = 'black-forest-labs/FLUX.1-dev'
 controlnet_model = 'YishaoAI/flux-dev-controlnet-canny-kid-clothes'
 
-controlnet = ControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.float16)
+controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.float16)
 pipe = FluxPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.float16)
 pipe.enable_model_cpu_offload()
 pipe.to("cuda")

requirements.txt

@@ -1,5 +1,5 @@
 torch
-diffusers
+git+https://github.com/huggingface/diffusers.git
 transformers
 accelerate
 controlnet_aux