test/test_inference.py

# -*- coding: UTF-8 -*-
"""
@Time : 28/05/2025 15:22
@Author : xiaoguangliang
@File : test_inference.py
@Project : Faice_text2face
"""
from inference_models.stable_diffusion_inference import inference_sd
from utils import timer

prompt = "Portrait of a young woman with long wavy hair, soft studio lighting, high contrast, 4k resolution, professional headshot"
# prompt = "Close-up of a smiling man with sharp jawline, cinematic lighting, shallow depth of field, bokeh background"

with timer("Test inference"):
    image, seed = inference_sd(prompt)
    image.save("test.png")