From e285af6e4817a34c86212d1b640aad7225f0c022 Mon Sep 17 00:00:00 2001
From: v0xie <28695009+v0xie@users.noreply.github.com>
Date: Fri, 19 Jul 2024 14:15:10 -0700
Subject: [PATCH] add beta schedule opts to xyz options

---
 scripts/xyz_grid.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
index 566493266..6a42a04d9 100644
--- a/scripts/xyz_grid.py
+++ b/scripts/xyz_grid.py
@@ -259,6 +259,8 @@ axis_options = [
     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")),