Update test_controlnet_lora_flux.py

RunpuWei · web-flow · commit cb9b331d6bcb · 2025-04-28T12:14:53.000+08:00
diff --git a/ControlNet_version/test_controlnet_lora_flux.py b/ControlNet_version/test_controlnet_lora_flux.py
@@ -20,7 +20,7 @@
 # Build pipeline
 controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
 transformer = FluxTransformer2DModel.from_pretrained(
-        "black-forest-labs/FLUX.1-deb", subfolder='transformer', torch_dtype=torch.bfloat16
+        "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
     )
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",

Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,7 @@`
`20`	`20`	`# Build pipeline`
`21`	`21`	`controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)`
`22`	`22`	`transformer = FluxTransformer2DModel.from_pretrained(`
`23`		`- "black-forest-labs/FLUX.1-deb", subfolder='transformer', torch_dtype=torch.bfloat16`
	`23`	`+ "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16`
`24`	`24`	`)`
`25`	`25`	`pipe = FluxControlNetInpaintingPipeline.from_pretrained(`
`26`	`26`	`"black-forest-labs/FLUX.1-dev",`