add beta schedule opts to xyz options

boin · Jul 19, 2024 · e285af6 · e285af6
1 parent f6f055a
commit e285af6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
@@ -259,6 +259,8 @@ def __init__(self, *args, **kwargs):
     AxisOption("Schedule min sigma", float, apply_override("sigma_min")),
     AxisOption("Schedule max sigma", float, apply_override("sigma_max")),
     AxisOption("Schedule rho", float, apply_override("rho")),
+    AxisOption("Beta schedule alpha", float, apply_override("beta_dist_alpha")),
+    AxisOption("Beta schedule beta", float, apply_override("beta_dist_beta")),
     AxisOption("Eta", float, apply_field("eta")),
     AxisOption("Clip skip", int, apply_override('CLIP_stop_at_last_layers')),
     AxisOption("Denoising", float, apply_field("denoising_strength")),