new ui

app.py

@@ -12,11 +12,11 @@ device="cuda"
 model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1"
 
 pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(device)
+pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(device)
+
 
 # pipe_inpaint = StableDiffusionInpaintPipeline.from_pretrained(model_id).to(device)    # work
 # pipe_inpaint = StableDiffusionInpaintPipeline(**pipe_text2img.components)   # not work
-pipe_img2img = StableDiffusionImg2ImgPipeline(**pipe_text2img.components).to(device)
-
 # def infer_text2img(prompt, guide, steps, width, height): 
 #     output = pipe_text2img(prompt, width=width, height=height, guidance_scale=guide, num_inference_steps=steps,)
 #     image = output.images[0]
@@ -51,21 +51,19 @@ with gr.Blocks() as demo:
                 ]
     with gr.Row():
         with gr.Column(scale=1, ):
-            with gr.Row(scale=0.5,):
-                with gr.Column():
-                    image_in = gr.Image(source='upload', elem_id="image_upload", type="pil", label="参考图(ref)")
-                with gr.Column():
-                    guide = gr.Slider(2, 15, value = 7, label = '文本引导强度(guidance scale)')
-                    steps = gr.Slider(10, 30, value = 20, step = 1, label = '迭代次数(inference steps)')
-                    width = gr.Slider(256, 768, value = 512, step = 64, label = '宽度(width)')
-                    height = gr.Slider(256, 768, value = 512, step = 64, label = '高度(height)')
-                    strength = gr.Slider(0, 1.0, value = 0.8, step = 0.05, label = '参考图改变程度(strength)')
+            image_out = gr.Image(label = '输出(output)')
+        with gr.Column(scale=1, ):
+            image_in = gr.Image(source='upload', elem_id="image_upload", type="pil", label="参考图（非必须）(ref)")
             prompt = gr.Textbox(label = '提示词(prompt)')
             submit_btn = gr.Button("生成图像(Generate)")
-        
-        with gr.Column(scale=1, ):
-            image_out = gr.Image(label = '输出(output)')
-            ex = gr.Examples(examples, fn=infer_text2img, inputs=[prompt, width, height], outputs=image_out)
+            with gr.Row(scale=0.5 ):
+                guide = gr.Slider(2, 15, value = 7, label = '文本引导强度(guidance scale)')
+                steps = gr.Slider(10, 30, value = 20, step = 1, label = '迭代次数(inference steps)')
+                width = gr.Slider(256, 768, value = 512, step = 64, label = '宽度(width)')
+                height = gr.Slider(256, 768, value = 512, step = 64, label = '高度(height)')
+                strength = gr.Slider(0, 1.0, value = 0.8, step = 0.05, label = '参考图改变程度(strength)')
+                ex = gr.Examples(examples, fn=infer_text2img, inputs=[prompt, guide, steps, width, height], outputs=image_out)
+
         # with gr.Column(scale=1, ):
         #     image_in = gr.Image(source='upload', tool='sketch', elem_id="image_upload", type="pil", label="Upload")
         #     inpaint_prompt = gr.Textbox(label = '提示词(prompt)')
@@ -75,4 +73,4 @@ with gr.Blocks() as demo:
         submit_btn.click(fn = infer_text2img, inputs = [prompt, guide, steps, width, height, image_in, strength], outputs = image_out)
         # inpaint_btn.click(fn = infer_inpaint, inputs = [inpaint_prompt, width, height, image_in], outputs = image_out)
         # img2img_btn.click(fn = infer_img2img, inputs = [img2img_prompt, width, height, image_in], outputs = image_out)
-demo.queue(concurrency_count=10).launch()
+demo.queue(concurrency_count=16).launch()