sonar lint fixes
This commit is contained in:
parent
c5eae688e3
commit
40e396a2c4
|
@ -35,7 +35,7 @@ def load_resrgan(
|
||||||
return cache_pipe
|
return cache_pipe
|
||||||
|
|
||||||
if not path.isfile(model_path):
|
if not path.isfile(model_path):
|
||||||
raise Exception("Real ESRGAN model not found at %s" % model_path)
|
raise FileNotFoundError("Real ESRGAN model not found at %s" % model_path)
|
||||||
|
|
||||||
if params.format == "onnx":
|
if params.format == "onnx":
|
||||||
# use ONNX acceleration, if available
|
# use ONNX acceleration, if available
|
||||||
|
@ -66,7 +66,7 @@ def load_resrgan(
|
||||||
scale=params.scale,
|
scale=params.scale,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
raise Exception("unknown platform %s" % params.format)
|
raise ValueError("unknown platform %s" % params.format)
|
||||||
|
|
||||||
dni_weight = None
|
dni_weight = None
|
||||||
if params.upscale_model == TAG_X4_V3 and params.denoise != 1:
|
if params.upscale_model == TAG_X4_V3 and params.denoise != 1:
|
||||||
|
|
|
@ -52,7 +52,7 @@ def process_tile_spiral(
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> Image.Image:
|
) -> Image.Image:
|
||||||
if scale != 1:
|
if scale != 1:
|
||||||
raise Exception("unsupported scale")
|
raise ValueError("unsupported scale")
|
||||||
|
|
||||||
width, height = source.size
|
width, height = source.size
|
||||||
image = Image.new("RGB", (width * scale, height * scale))
|
image = Image.new("RGB", (width * scale, height * scale))
|
||||||
|
|
|
@ -30,7 +30,7 @@ def mask_filter_gaussian_multiply(
|
||||||
"""
|
"""
|
||||||
noise = mask_filter_none(mask, dims, origin)
|
noise = mask_filter_none(mask, dims, origin)
|
||||||
|
|
||||||
for i in range(rounds):
|
for _i in range(rounds):
|
||||||
blur = noise.filter(ImageFilter.GaussianBlur(5))
|
blur = noise.filter(ImageFilter.GaussianBlur(5))
|
||||||
noise = ImageChops.multiply(noise, blur)
|
noise = ImageChops.multiply(noise, blur)
|
||||||
|
|
||||||
|
@ -45,7 +45,7 @@ def mask_filter_gaussian_screen(
|
||||||
"""
|
"""
|
||||||
noise = mask_filter_none(mask, dims, origin)
|
noise = mask_filter_none(mask, dims, origin)
|
||||||
|
|
||||||
for i in range(rounds):
|
for _i in range(rounds):
|
||||||
blur = noise.filter(ImageFilter.GaussianBlur(5))
|
blur = noise.filter(ImageFilter.GaussianBlur(5))
|
||||||
noise = ImageChops.screen(noise, blur)
|
noise = ImageChops.screen(noise, blur)
|
||||||
|
|
||||||
|
@ -88,7 +88,7 @@ def noise_source_gaussian(
|
||||||
noise = noise_source_uniform(source, dims, origin)
|
noise = noise_source_uniform(source, dims, origin)
|
||||||
noise.paste(source, origin)
|
noise.paste(source, origin)
|
||||||
|
|
||||||
for i in range(rounds):
|
for _i in range(rounds):
|
||||||
noise = noise.filter(ImageFilter.GaussianBlur(5))
|
noise = noise.filter(ImageFilter.GaussianBlur(5))
|
||||||
|
|
||||||
return noise
|
return noise
|
||||||
|
|
|
@ -803,8 +803,8 @@ def ready():
|
||||||
done, progress = executor.done(output_file)
|
done, progress = executor.done(output_file)
|
||||||
|
|
||||||
if done is None:
|
if done is None:
|
||||||
file = base_join(context.output_path, output_file)
|
output = base_join(context.output_path, output_file)
|
||||||
if path.exists(file):
|
if path.exists(output):
|
||||||
return ready_reply(True)
|
return ready_reply(True)
|
||||||
|
|
||||||
return ready_reply(done, progress=progress)
|
return ready_reply(done, progress=progress)
|
||||||
|
|
|
@ -44,7 +44,7 @@ class JobContext:
|
||||||
with self.device_index.get_lock():
|
with self.device_index.get_lock():
|
||||||
device_index = self.device_index.value
|
device_index = self.device_index.value
|
||||||
if device_index < 0:
|
if device_index < 0:
|
||||||
raise Exception("job has not been assigned to a device")
|
raise ValueError("job has not been assigned to a device")
|
||||||
else:
|
else:
|
||||||
device = self.devices[device_index]
|
device = self.devices[device_index]
|
||||||
logger.debug("job %s assigned to device %s", self.key, device)
|
logger.debug("job %s assigned to device %s", self.key, device)
|
||||||
|
@ -57,7 +57,7 @@ class JobContext:
|
||||||
def on_progress(step: int, timestep: int, latents: Any):
|
def on_progress(step: int, timestep: int, latents: Any):
|
||||||
on_progress.step = step
|
on_progress.step = step
|
||||||
if self.is_cancelled():
|
if self.is_cancelled():
|
||||||
raise Exception("job has been cancelled")
|
raise RuntimeError("job has been cancelled")
|
||||||
else:
|
else:
|
||||||
logger.debug("setting progress for job %s to %s", self.key, step)
|
logger.debug("setting progress for job %s to %s", self.key, step)
|
||||||
self.set_progress(step)
|
self.set_progress(step)
|
||||||
|
|
|
@ -65,7 +65,7 @@ def get_not_empty(args: Any, key: str, default: Any) -> Any:
|
||||||
return val
|
return val
|
||||||
|
|
||||||
|
|
||||||
def get_size(val: Union[int, str, None]) -> SizeChart:
|
def get_size(val: Union[int, str, None]) -> Union[int, SizeChart]:
|
||||||
if val is None:
|
if val is None:
|
||||||
return SizeChart.auto
|
return SizeChart.auto
|
||||||
|
|
||||||
|
@ -79,7 +79,7 @@ def get_size(val: Union[int, str, None]) -> SizeChart:
|
||||||
|
|
||||||
return int(val)
|
return int(val)
|
||||||
|
|
||||||
raise Exception("invalid size")
|
raise ValueError("invalid size")
|
||||||
|
|
||||||
|
|
||||||
def run_gc(devices: List[DeviceParams] = None):
|
def run_gc(devices: List[DeviceParams] = None):
|
||||||
|
|
|
@ -4,11 +4,11 @@ test_images=0
|
||||||
while true;
|
while true;
|
||||||
do
|
do
|
||||||
curl "http://${test_host}:5000/api/txt2img?"\
|
curl "http://${test_host}:5000/api/txt2img?"\
|
||||||
'cfg=16.00&steps=35&scheduler=deis-multi&seed=-1&'\
|
'cfg=16.00&steps=35&scheduler=deis-multi&seed=-1&'\
|
||||||
'prompt=an+astronaut+eating+a+hamburger&negativePrompt=&'\
|
'prompt=an+astronaut+eating+a+hamburger&negativePrompt=&'\
|
||||||
'model=stable-diffusion-v1-5&platform=any&'\
|
'model=stable-diffusion-onnx-v1-5&platform=any&'\
|
||||||
'upscaling=upscaling-real-esrgan-x2-plus&correction=correction-codeformer&'\
|
'upscaling=upscaling-real-esrgan-x2-plus&correction=correction-codeformer&'\
|
||||||
'lpw=false&width=512&height=512&upscaleOrder=correction-both' \
|
'lpw=false&width=512&height=512&upscaleOrder=correction-both' \
|
||||||
-X 'POST' \
|
-X 'POST' \
|
||||||
--compressed \
|
--compressed \
|
||||||
--insecure || break;
|
--insecure || break;
|
||||||
|
|
Loading…
Reference in New Issue