wanghaofan
commited on
Commit
•
b4f6968
1
Parent(s):
c0544ab
Update README.md
Browse files
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)
|