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