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)
|