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:
Chenlei Hu
2024-02-09 21:33:05 +00:00
committed by GitHub
parent 472a510151
commit ac4a8820a5
6 changed files with 89 additions and 1 deletions

View File

@@ -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):

View File

@@ -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

View File

@@ -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