Unify CodeFormer and GFPGAN restoration backends, use Spandrel for GFPGAN

This commit is contained in:
Aarni Koskela
2023-12-25 23:01:02 +02:00
parent b0f5934234
commit b621a63cf6
11 changed files with 302 additions and 228 deletions

View File

@@ -1,10 +1,16 @@
import base64
import os
import pytest
import base64
test_files_path = os.path.dirname(__file__) + "/test_files"
test_outputs_path = os.path.dirname(__file__) + "/test_outputs"
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")
def file_to_base64(filename):
@@ -23,3 +29,8 @@ def img2img_basic_image_base64() -> str:
@pytest.fixture(scope="session") # session so we don't read this over and over
def mask_basic_image_base64() -> str:
return file_to_base64(os.path.join(test_files_path, "mask_basic.png"))
@pytest.fixture(scope="session")
def initialize() -> None:
import webui # noqa: F401