From b36b465e0541b3309887158fa62e749b0533b3e3 Mon Sep 17 00:00:00 2001 From: Siddharth Venkatesan Date: Sun, 11 Dec 2022 23:13:26 -0800 Subject: [PATCH] Import some dependencies only for stable diffusion test (#391) --- tests/integration/llm/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/llm/client.py b/tests/integration/llm/client.py index 6b32e94b2..40c28c79d 100644 --- a/tests/integration/llm/client.py +++ b/tests/integration/llm/client.py @@ -3,8 +3,6 @@ import subprocess as sp import logging import math -from PIL import Image -from io import BytesIO logging.basicConfig(level=logging.INFO) parser = argparse.ArgumentParser(description='Build the LLM configs') @@ -188,6 +186,8 @@ def test_sd_handler(model, model_spec): spec = sd_model_spec[model] if "worker" in spec: check_worker_number(spec["worker"]) + from PIL import Image + from io import BytesIO for size in spec["size"]: for step in spec["steps"]: req = {"prompt": "A bird and cat flying through space"}