mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-04-27 09:41:31 +00:00
Fix CQ tests (#141)
* Make test client run on cpu * test on cpu try fix device try fix device try fix device * Use real SD1.5 model for testing * ckpt nits * Remove coverage calls
This commit is contained in:
@@ -11,6 +11,7 @@ def pytest_configure(config):
|
||||
# We don't want to fail on Py.test command line arguments being
|
||||
# parsed by webui:
|
||||
os.environ.setdefault("IGNORE_CMD_ARGS_ERRORS", "1")
|
||||
os.environ.setdefault("FORGE_CQ_TEST", "1")
|
||||
|
||||
|
||||
def file_to_base64(filename):
|
||||
|
||||
@@ -8,6 +8,7 @@ from PIL import Image
|
||||
|
||||
@pytest.mark.usefixtures("initialize")
|
||||
@pytest.mark.parametrize("restorer_name", ["gfpgan", "codeformer"])
|
||||
@pytest.mark.skip # Skip for forge.
|
||||
def test_face_restorers(restorer_name):
|
||||
from modules import shared
|
||||
|
||||
|
||||
@@ -69,6 +69,7 @@ def test_txt2img_with_tiling_performed(url_txt2img, simple_txt2img_request):
|
||||
assert requests.post(url_txt2img, json=simple_txt2img_request).status_code == 200
|
||||
|
||||
|
||||
@pytest.mark.skip # Skip for forge.
|
||||
def test_txt2img_with_restore_faces_performed(url_txt2img, simple_txt2img_request):
|
||||
simple_txt2img_request["restore_faces"] = True
|
||||
assert requests.post(url_txt2img, json=simple_txt2img_request).status_code == 200
|
||||
|
||||
Reference in New Issue
Block a user