From 9dce2aa735d2b52ce3966efdb3f4a22c356a6bb4 Mon Sep 17 00:00:00 2001
From: w-e-w <40751091+w-e-w@users.noreply.github.com>
Date: Mon, 21 Aug 2023 23:07:09 +0900
Subject: add RNG source to XYZ

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

(limited to 'scripts')

diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
index 3d9fe691..daaf761f 100644
--- a/scripts/xyz_grid.py
+++ b/scripts/xyz_grid.py
@@ -267,6 +267,7 @@ axis_options = [
     AxisOption("Always discard next-to-last sigma", str, apply_override('always_discard_next_to_last_sigma', boolean=True), choices=boolean_choice(reverse=True)),
     AxisOption("Refiner checkpoint", str, apply_field('refiner_checkpoint'), format_value=format_remove_path, confirm=confirm_checkpoints_or_none, cost=1.0, choices=lambda: ['None'] + sorted(sd_models.checkpoints_list, key=str.casefold)),
     AxisOption("Refiner switch at", float, apply_field('refiner_switch_at')),
+    AxisOption("RNG source", str, apply_override("randn_source"), choices=lambda: ["GPU", "CPU", "NV"]),
 ]
 
 
-- 
cgit v1.2.1