From 4d34868ad2e69849cb8953784e1d5b8f7ef40ef7 Mon Sep 17 00:00:00 2001 From: huchenlei Date: Sun, 21 Apr 2024 08:47:14 -0400 Subject: [PATCH] Update xyz module detecction --- scripts/animatediff_xyz.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/animatediff_xyz.py b/scripts/animatediff_xyz.py index 0fe9baa0..ae27af55 100644 --- a/scripts/animatediff_xyz.py +++ b/scripts/animatediff_xyz.py @@ -119,7 +119,7 @@ def choices_bool(): def find_xyz_module() -> Optional[ModuleType]: for data in scripts.scripts_data: - if data.script_class.__module__ in {"xyz_grid.py", "xy_grid.py"} and hasattr(data, "module"): + if data.script_class.__module__ in {"xyz_grid.py", "xy_grid.py", "scripts.xyz_grid", "scripts.xy_grid"} and hasattr(data, "module"): return data.module return None