Skip to content

Commit

Permalink
Update xyz module detecction (#512)
Browse files Browse the repository at this point in the history
  • Loading branch information
huchenlei authored Apr 22, 2024
1 parent ecc6cf7 commit 856fc0e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/animatediff_xyz.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 856fc0e

Please sign in to comment.