Merge pull request #4635 from mezotaken/master

name: Run basic features tests on CPU with empty SD model

on:

  - push

  - pull_request

jobs:

  test:

    runs-on: ubuntu-latest

    steps:

      - name: Checkout Code

        uses: actions/checkout@v3

      - name: Set up Python 3.10

        uses: actions/setup-python@v4

        with:

          python-version: 3.10.6

      - uses: actions/cache@v3

        with:

          path: ~/.cache/pip

          key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements.txt') }}

          restore-keys: ${{ runner.os }}-pip-

      - name: Run tests

        run: python launch.py --tests basic_features --no-half --disable-opt-split-attention --use-cpu all --skip-torch-cuda-test

      - name: Upload main app stdout-stderr

        uses: actions/upload-artifact@v3

        if: always()

        with:

          name: stdout-stderr

          path: |

            test/stdout.txt

            test/stderr.txt

    return [x for x in args if x != name], name in args

def extract_opt(args, name):

    opt = None

    is_present = False

    if name in args:

        is_present = True

        idx = args.index(name)

        del args[idx]

        if idx < len(args) and args[idx][0] != "-":

            opt = args[idx]

            del args[idx]

    return args, is_present, opt

def run(command, desc=None, errdesc=None, custom_env=None):

    if desc is not None:

        print(desc)

    blip_commit_hash = os.environ.get('BLIP_COMMIT_HASH', "48211a1594f1321b00f14c9f7a5b4813144b2fb9")

    sys.argv += shlex.split(commandline_args)

    test_argv = [x for x in sys.argv if x != '--tests']

    sys.argv, skip_torch_cuda_test = extract_arg(sys.argv, '--skip-torch-cuda-test')

    sys.argv, reinstall_xformers = extract_arg(sys.argv, '--reinstall-xformers')

    sys.argv, update_check = extract_arg(sys.argv, '--update-check')

    sys.argv, run_tests = extract_arg(sys.argv, '--tests')

    sys.argv, run_tests, test_dir = extract_opt(sys.argv, '--tests')

    xformers = '--xformers' in sys.argv

    ngrok = '--ngrok' in sys.argv

        exit(0)

    if run_tests:

        tests(test_argv)

        exit(0)

        exitcode = tests(test_dir)

        exit(exitcode)

def tests(argv):

    if "--api" not in argv:

        argv.append("--api")

def tests(test_dir):

    if "--api" not in sys.argv:

        sys.argv.append("--api")

    if "--ckpt" not in sys.argv:

        sys.argv.append("--ckpt")

        sys.argv.append("./test/test_files/empty.pt")

    if "--skip-torch-cuda-test" not in sys.argv:

        sys.argv.append("--skip-torch-cuda-test")

    print(f"Launching Web UI in another process for testing with arguments: {' '.join(argv[1:])}")

    print(f"Launching Web UI in another process for testing with arguments: {' '.join(sys.argv[1:])}")

    with open('test/stdout.txt', "w", encoding="utf8") as stdout, open('test/stderr.txt', "w", encoding="utf8") as stderr:

        proc = subprocess.Popen([sys.executable, *argv], stdout=stdout, stderr=stderr)

        proc = subprocess.Popen([sys.executable, *sys.argv], stdout=stdout, stderr=stderr)

    import test.server_poll

    test.server_poll.run_tests()

    exitcode = test.server_poll.run_tests(proc, test_dir)

    print(f"Stopping Web UI process with id {proc.pid}")

    proc.kill()

    return exitcode

def start():

            "codeformer_visibility": 0,

            "codeformer_weight": 0,

            "upscaling_resize": 2,

            "upscaling_resize_w": 512,

            "upscaling_resize_h": 512,

            "upscaling_resize_w": 128,

            "upscaling_resize_h": 128,

            "upscaling_crop": True,

            "upscaler_1": "None",

            "upscaler_2": "None",

import unittest

import requests

class TestTxt2ImgWorking(unittest.TestCase):

    def setUp(self):

        self.url_txt2img = "http://localhost:7860/sdapi/v1/txt2img"

        self.simple_txt2img = {

            "enable_hr": False,

            "denoising_strength": 0,

            "firstphase_width": 0,

            "firstphase_height": 0,

            "prompt": "example prompt",

            "styles": [],

            "seed": -1,

            "subseed": -1,

            "subseed_strength": 0,

            "seed_resize_from_h": -1,

            "seed_resize_from_w": -1,

            "batch_size": 1,

            "n_iter": 1,

            "steps": 3,

            "cfg_scale": 7,

            "width": 64,

            "height": 64,

            "restore_faces": False,

            "tiling": False,

            "negative_prompt": "",

            "eta": 0,

            "s_churn": 0,

            "s_tmax": 0,

            "s_tmin": 0,

            "s_noise": 1,

            "sampler_index": "Euler a"

        }

    def test_txt2img_with_restore_faces_performed(self):

        self.simple_txt2img["restore_faces"] = True

        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)

class TestTxt2ImgCorrectness(unittest.TestCase):

    pass

if __name__ == "__main__":

    unittest.main()