From dbf40d18346ff9993bdd70886c55d34cf383e16b Mon Sep 17 00:00:00 2001 From: s9roll7 Date: Wed, 13 Dec 2023 20:35:51 +0900 Subject: [PATCH] fix #199 (controlnet_normalbae) --- src/animatediff/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/animatediff/generate.py b/src/animatediff/generate.py index 8b3defcb..a7e223b8 100644 --- a/src/animatediff/generate.py +++ b/src/animatediff/generate.py @@ -1040,7 +1040,7 @@ def controlnet_preprocess( save_path = det_dir.joinpath(f"{org_name}.png") ref_image.save(save_path) - controlnet_no_shrink = ["controlnet_tile","animatediff_controlnet","controlnet_canny","controlnet_depth","controlnet_lineart","controlnet_lineart_anime","controlnet_scribble","controlnet_seg","controlnet_softedge","controlnet_mlsd"] + controlnet_no_shrink = ["controlnet_tile","animatediff_controlnet","controlnet_canny","controlnet_normalbae","controlnet_depth","controlnet_lineart","controlnet_lineart_anime","controlnet_scribble","controlnet_seg","controlnet_softedge","controlnet_mlsd"] if "no_shrink_list" in controlnet_map: controlnet_no_shrink = controlnet_map["no_shrink_list"]