1580158
/
stable-diffusion-webui-randomize
mirror of https://github.com/innightwolfsleep/stable-diffusion-webui-randomize.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cleanup: Better variable names

pull/6/head
MMaker 2022-11-09 19:41:55 -05:00
parent 1b9787a57d
commit f82a376987
No known key found for this signature in database
GPG Key ID: CCE79B8FEDA40FB2
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/randomize.py
View File

@ -44,10 +44,10 @@ class RandomizeScript(scripts.Script):
def _opt(self, opt, p, prefix='randomize_param_'): def _opt(self, opt, p, prefix='randomize_param_'):
opt_name = f'{prefix}{opt}' opt_name = f'{prefix}{opt}'
option: str = getattr(shared.opts, opt_name) opt_val: str = getattr(shared.opts, opt_name)
split: list[str] = option.split(',') opt_arr: list[str] = opt_val.split(',')
if split[0].isdigit(): if opt_arr[0].isdigit():
vals = [float(v) for v in split] vals = [float(v) for v in opt_arr]
rand = self._rand(vals[0], vals[1], vals[2]) rand = self._rand(vals[0], vals[1], vals[2])
if rand.is_integer(): if rand.is_integer():
return int(rand) return int(rand)
@ -55,9 +55,9 @@ class RandomizeScript(scripts.Script):
return float(rand) return float(rand)
else: else:
if opt == 'sampler_index': if opt == 'sampler_index':
return build_samplers_dict(p).get(random.choice(split).lower(), None) return build_samplers_dict(p).get(random.choice(opt_arr).lower(), None)
else: else:
return random.choice(split) return random.choice(opt_arr)
def _rand(self, start: float, stop: float, step: float) -> float: def _rand(self, start: float, stop: float, step: float) -> float:
return random.randint(0, int((stop - start) / step)) * step + start return random.randint(0, int((stop - start) / step)) * step + start