Merge pull request #1 from s9roll7/main

fix s9roll7#199 (controlnet_normalbae)
hassantsyed · Dec 14, 2023 · 6e6b37e · 6e6b37e
2 parents c744817 + dbf40d1
commit 6e6b37e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"]