feat: add test

app.py

@@ -22,6 +22,9 @@ examples = [
 
 css = """
 body {
+    position: fixed;
+    top: 0;
+    left: 0;
     background-image: url('https://lh3.googleusercontent.com/d/1y7H4WIjnBAcNvvi-3qOV_ORE-jMXP4fr');
     background-repeat: no-repeat;
     background-attachment: fixed;

stable_diffusion_inference.py

@@ -18,15 +18,17 @@ from utils import timer
 
 model_path = 'Ngene787/Faice_text2face'
 
-accelerator = Accelerator(mixed_precision="fp16", gradient_accumulation_steps=1)
+if torch.backends.mps.is_available():
+    accelerator = Accelerator(gradient_accumulation_steps=1)
+else:
+    accelerator = Accelerator(mixed_precision="fp16", gradient_accumulation_steps=1)
+
 logger.info("Loading model ...")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 if torch.cuda.is_available():
     torch_dtype = torch.float16
 else:
     torch_dtype = torch.float32
-# device = "cpu"
-# torch_dtype = torch.float32
 pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch_dtype,
                                                low_cpu_mem_usage=True,
                                                # requires_safety_checker=False
@@ -37,11 +39,6 @@ pipe = accelerator.prepare(pipe)
 # Enable memory-efficient attention
 # pipe.enable_xformers_memory_efficient_attention()
 
-# Enable attention slicing
-# pipe.enable_attention_slicing()
-
-# Enable VAE slicing
-# pipe.enable_vae_slicing()
 
 MAX_SEED = np.iinfo(np.int32).max
 

test/test_api.py

@@ -0,0 +1,23 @@
+# -*- coding: UTF-8 -*-
+"""
+@Time : 30/05/2025 10:18
+@Author : xiaoguangliang
+@File : test_api.py
+@Project : Faice_text2face
+"""
+from gradio_client import Client
+
+client = Client("https://e517e907a9e4213655.gradio.live/")
+
+prompt = "Portrait of a young woman with long wavy hair, soft studio lighting, high contrast, 4k resolution, professional headshot"
+
+result = client.predict(
+    prompt=prompt,
+    negative_prompt="",
+    seed=0,
+    randomize_seed=False,
+    guidance_scale=7.5,
+    num_inference_steps=100,
+    api_name="/inference"
+)
+print(result)

test.py → test/test_inference.py

@@ -2,7 +2,7 @@
 """
 @Time : 28/05/2025 15:22
 @Author : xiaoguangliang
-@File : test.py
+@File : test_inference.py
 @Project : Faice_text2face
 """
 from stable_diffusion_inference import inference