File size: 578 Bytes
2e3190d
 
 
 
592a5b4
2e3190d
 
4d24ceb
453c092
2e3190d
51910ed
 
2e3190d
d082263
ae53881
64b93e7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# -*- 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")