mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
feat(ui): filter generated sampler and upscaler lists
This commit is contained in:
@@ -159,9 +159,11 @@ def images_tensor_to_samples(image, approximation=None, model=None): # pylint: d
|
||||
return x_latent
|
||||
|
||||
|
||||
def get_sampler_name(sampler_index: int | None = None, img: bool = False) -> str:
|
||||
def get_sampler_name(sampler_index: int | str | None = None, img: bool = False) -> str:
|
||||
sampler_index = sampler_index or 0
|
||||
if len(sd_samplers.samplers) > sampler_index:
|
||||
if isinstance(sampler_index, str) and any(sampler.name == sampler_index for sampler in sd_samplers.samplers):
|
||||
sampler_name = sampler_index
|
||||
elif isinstance(sampler_index, int) and 0 <= sampler_index < len(sd_samplers.samplers):
|
||||
sampler_name = sd_samplers.samplers[sampler_index].name
|
||||
else:
|
||||
sampler_name = "Default"
|
||||
|
||||
Reference in New Issue
Block a user