Spaces:
Running on Zero
Running on Zero
File size: 2,595 Bytes
5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 5ac741c 76862de 213bf15 76862de 5ac741c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | import pytest
pytestmark = pytest.mark.gpu
@pytest.fixture(scope="module")
def real_backend():
"""Build a real backend with real weights. ~30 GB download on first run."""
import backend
return backend.ZImageStudioBackend()
def test_t2i_turbo_produces_image(real_backend):
from PIL import Image
image, meta = real_backend.generate(
mode="t2i",
params=dict(
prompt="a red apple on a wooden table",
negative_prompt="",
model="Turbo",
steps=8,
cfg=1.0,
width=384,
height=384,
seed=42,
lora_path=None,
lora_strength=0.0,
),
)
assert isinstance(image, Image.Image)
assert image.size == (384, 384)
assert meta["model"] == "Turbo"
def test_t2i_base_produces_image(real_backend):
from PIL import Image
image, _meta = real_backend.generate(
mode="t2i",
params=dict(
prompt="a red apple on a wooden table",
negative_prompt="blurry",
model="Base",
steps=15,
cfg=4.0,
width=384,
height=384,
seed=42,
lora_path=None,
lora_strength=0.0,
),
)
assert isinstance(image, Image.Image)
def test_controlnet_produces_image(real_backend):
import numpy as np
from PIL import Image
arr = np.random.randint(0, 255, (384, 384, 3), dtype=np.uint8)
image, _meta = real_backend.generate(
mode="controlnet",
params=dict(
prompt="a portrait of a person, dramatic light",
input_image=Image.fromarray(arr),
preprocessor="Canny",
controlnet_scale=1.0,
steps=9,
seed=42,
lora_path=None,
lora_strength=0.0,
),
)
assert isinstance(image, Image.Image)
def test_upscale_produces_image(real_backend, tmp_path):
import numpy as np
from huggingface_hub import hf_hub_download
from PIL import Image
arr = np.random.randint(0, 255, (256, 256, 3), dtype=np.uint8)
image, _meta = real_backend.generate(
mode="upscale",
params=dict(
prompt="masterpiece, 8k",
input_image=Image.fromarray(arr),
refine_steps=5,
refine_denoise=0.33,
seed=42,
lora_path=None,
lora_strength=0.0,
esrgan_model_path=hf_hub_download("lllyasviel/Annotators", "RealESRGAN_x4plus.pth"),
),
)
assert image.size == (512, 512)
|