sonar lint
This commit is contained in:
parent
1818a36c11
commit
c134edf4b3
|
@ -28,9 +28,7 @@ class UpscaleOutpaintStageTests(unittest.TestCase):
|
||||||
3,
|
3,
|
||||||
0.1,
|
0.1,
|
||||||
),
|
),
|
||||||
ServerContext(
|
ServerContext(),
|
||||||
# model_path="../models",
|
|
||||||
),
|
|
||||||
None,
|
None,
|
||||||
ImageParams(
|
ImageParams(
|
||||||
"../models/stable-diffusion-onnx-v1-inpainting",
|
"../models/stable-diffusion-onnx-v1-inpainting",
|
||||||
|
|
|
@ -28,9 +28,7 @@ class UpscaleSwinIRStageTests(unittest.TestCase):
|
||||||
3,
|
3,
|
||||||
0.1,
|
0.1,
|
||||||
),
|
),
|
||||||
ServerContext(
|
ServerContext(),
|
||||||
# model_path="../models",
|
|
||||||
),
|
|
||||||
None,
|
None,
|
||||||
None,
|
None,
|
||||||
sources,
|
sources,
|
||||||
|
|
|
@ -18,7 +18,7 @@ from tests.helpers import test_device
|
||||||
|
|
||||||
|
|
||||||
def main_memory(_worker):
|
def main_memory(_worker):
|
||||||
raise Exception(MEMORY_ERRORS[0])
|
raise MemoryError(MEMORY_ERRORS[0])
|
||||||
|
|
||||||
|
|
||||||
def main_retry(_worker):
|
def main_retry(_worker):
|
||||||
|
|
Loading…
Reference in New Issue