Fix xyz module name (#99)

mcmonkeyprojects · Apr 21, 2024 · 701a208 · 701a208
1 parent d042bb0
commit 701a208
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/dynamic_thresholding.py b/scripts/dynamic_thresholding.py
@@ -234,7 +234,7 @@ def combine_denoised(self, x_out, conds_list, uncond, cond_scale):
 ######################### XYZ Plot Script Support logic #########################
 
 def make_axis_options():
-    xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module
+    xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ in ("xyz_grid.py", "scripts.xyz_grid")][0].module
     def apply_mimic_scale(p, x, xs):
         if x != 0:
             setattr(p, "dynthres_enabled", True)