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: for cat in params_and_values:
choices = list(params_and_values[cat].keys()) choices = list(params_and_values[cat].keys())
if shared.args.loader is not None: 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: if len(choices) > 0:
choice = random.choice(choices) choice = random.choice(choices)