Shakker-Labs
/

FLUX.1-dev-ControlNet-Depth

image-generation

Stable Diffusion

Model card Files Files and versions

wanghaofan commited on Aug 27, 2024

Commit

2fdfbcf

·

verified ·

1 Parent(s): d48aa11

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -44,8 +44,8 @@ import torch
 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel
-controlnet_model = "black-forest-labs/FLUX.1-dev"
-base_model = "Shakker-Labs/FLUX.1-dev-ControlNet-Depth"
 controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 pipe = FluxControlNetPipeline.from_pretrained(

 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel
+base_model = "black-forest-labs/FLUX.1-dev"
+controlnet_model = "Shakker-Labs/FLUX.1-dev-ControlNet-Depth"
 controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 pipe = FluxControlNetPipeline.from_pretrained(