From a10487986925ca8fd07ee7ae7fc5034752298551 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Mon, 22 May 2023 21:52:46 +0800
Subject: Add custom karras scheduler

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

(limited to 'scripts/xyz_grid.py')

diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
index da820b39..6ea049ee 100644
--- a/scripts/xyz_grid.py
+++ b/scripts/xyz_grid.py
@@ -220,6 +220,9 @@ axis_options = [
     AxisOption("Sigma min", float, apply_field("s_tmin")),
     AxisOption("Sigma max", float, apply_field("s_tmax")),
     AxisOption("Sigma noise", float, apply_field("s_noise")),
+    AxisOption("Karras Scheduler Sigma Min", float, apply_field("sigma_min")),
+    AxisOption("Karras Scheduler Sigma Max", float, apply_field("sigma_max")),
+    AxisOption("Karras Scheduler rho", float, apply_field("rho")),
     AxisOption("Eta", float, apply_field("eta")),
     AxisOption("Clip skip", int, apply_clip_skip),
     AxisOption("Denoising", float, apply_field("denoising_strength")),
-- 
cgit v1.2.3