This commit is contained in:
oobabooga 2024-02-13 16:05:41 -08:00
parent 193548edce
commit 069ed7c6ef
2 changed files with 5 additions and 5 deletions

View File

@ -125,7 +125,7 @@ def random_preset(state):
for cat in params_and_values:
choices = list(params_and_values[cat].keys())
if shared.args.loader is not None:
choices = [x for x in choices if loader_contains(sampler)]
choices = [x for x in choices if loader_contains(x)]
if len(choices) > 0:
choice = random.choice(choices)