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

Better error handling

pull/6/head
MMaker 2022-11-11 10:49:08 -05:00
parent 075fde56b3
commit f55fa7d4e6
No known key found for this signature in database
GPG Key ID: CCE79B8FEDA40FB2
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/randomize.py
View File

@ -24,7 +24,9 @@ class RandomizeScript(scripts.Script):
opt = self._opt(param, p) opt = self._opt(param, p)
if opt is not None: if opt is not None:
setattr(p, param, opt) setattr(p, param, opt)
except TypeError: else:
print(f'Skipping randomizing param `{param}` -- incorrect value')
except (TypeError, IndexError):
print(f'Failed to randomize param `{param}` -- incorrect value?') print(f'Failed to randomize param `{param}` -- incorrect value?')
# Other params # Other params
@ -45,7 +47,7 @@ class RandomizeScript(scripts.Script):
setattr(p, 'truncate_y', 0) setattr(p, 'truncate_y', 0)
setattr(p, 'denoising_strength', self._opt('denoising_strength', p, prefix='randomize_hires_')) setattr(p, 'denoising_strength', self._opt('denoising_strength', p, prefix='randomize_hires_'))
except TypeError: except (TypeError, IndexError):
print(f'Failed to utilize highres. fix -- incorrect value?') print(f'Failed to utilize highres. fix -- incorrect value?')
else: else:
return return
@ -70,7 +72,7 @@ class RandomizeScript(scripts.Script):
if opt == 'sampler_index': if opt == 'sampler_index':
return build_samplers_dict(p).get(random.choice(opt_arr).lower(), None) return build_samplers_dict(p).get(random.choice(opt_arr).lower(), None)
else: else:
return random.choice(opt_arr) return None
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