JiantaoLin
commited on
Commit
·
3ddd275
1
Parent(s):
695a6b1
new
Browse files
pipeline/kiss3d_wrapper.py
CHANGED
@@ -76,7 +76,7 @@ def init_wrapper_from_config(config_path):
|
|
76 |
flux_pipe = FluxImg2ImgPipeline.from_pretrained(flux_base_model_pth, torch_dtype=dtype_[flux_dtype], token=access_token)
|
77 |
|
78 |
# load flux model and controlnet
|
79 |
-
if flux_controlnet_pth is not None:
|
80 |
flux_controlnet = FluxControlNetModel.from_pretrained(flux_controlnet_pth, torch_dtype=torch.bfloat16)
|
81 |
flux_pipe = convert_flux_pipeline(flux_pipe, FluxControlNetImg2ImgPipeline, controlnet=[flux_controlnet])
|
82 |
|
|
|
76 |
flux_pipe = FluxImg2ImgPipeline.from_pretrained(flux_base_model_pth, torch_dtype=dtype_[flux_dtype], token=access_token)
|
77 |
|
78 |
# load flux model and controlnet
|
79 |
+
if flux_controlnet_pth is not None and False:
|
80 |
flux_controlnet = FluxControlNetModel.from_pretrained(flux_controlnet_pth, torch_dtype=torch.bfloat16)
|
81 |
flux_pipe = convert_flux_pipeline(flux_pipe, FluxControlNetImg2ImgPipeline, controlnet=[flux_controlnet])
|
82 |
|