diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py index 3d9fe6913..daaf761f1 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"]), ]