wanghaofan commited on
Commit
b4f6968
1 Parent(s): c0544ab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -23,6 +23,7 @@ from diffusers.utils import load_image
23
  from diffusers.pipelines.flux.pipeline_flux_controlnet import FluxControlNetPipeline
24
  from diffusers.models.controlnet_flux import FluxControlNetModel
25
 
 
26
  controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Canny-alpha'
27
  controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
28
  pipe = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)
 
23
  from diffusers.pipelines.flux.pipeline_flux_controlnet import FluxControlNetPipeline
24
  from diffusers.models.controlnet_flux import FluxControlNetModel
25
 
26
+ base_model = 'black-forest-labs/FLUX.1-dev'
27
  controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Canny-alpha'
28
  controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
29
  pipe = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)