feat: add test for inference and remove the picture.

app.py

@@ -7,9 +7,11 @@
 """
 import gradio as gr
 
-from stable_diffusion_inference import inference, MAX_SEED, MAX_IMAGE_SIZE
+from stable_diffusion_inference import inference, MAX_SEED
 from utils import timer
 
+MAX_IMAGE_SIZE = 1024
+
 examples = [
     "Portrait of a young woman with long wavy hair, soft studio lighting, high contrast, 4k resolution, professional headshot",
     "Close-up of a smiling man with sharp jawline, cinematic lighting, shallow depth of field, bokeh background",

stable_diffusion_inference.py

@@ -41,7 +41,6 @@ pipe.enable_attention_slicing()
 pipe.enable_vae_slicing()
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 1024
 
 
 @spaces.GPU(duration=65)
@@ -49,10 +48,8 @@ def inference(prompt,
               negative_prompt="",
               seed=0,
               randomize_seed=False,
-              width=MAX_IMAGE_SIZE,
-              height=MAX_IMAGE_SIZE,
               guidance_scale=7.5,
-              num_inference_steps=10,
+              num_inference_steps=50,
               progress=gr.Progress(track_tqdm=True), ):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -66,8 +63,6 @@ def inference(prompt,
         guidance_scale=guidance_scale,
         eta=0.0,
         num_inference_steps=num_inference_steps,
-        # width=width,
-        # height=height,
         generator=generator,
     ).images[0]
     # image = pipe(prompt).images[0]