Update app.py from anycoder

app.py

@@ -5,29 +5,35 @@ from diffusers import NewbiePipeline
 from transformers import AutoModel
 import random
 import warnings
+from diffusers import DPMSolverMultistepScheduler
 
 warnings.filterwarnings("ignore")
 
-model_path = "Disty0/NewBie-image-Exp0.1-Diffusers"  # NewBie-AI/NewBie-image-Exp0.1
+# 优化：使用 float16 以获得更好的 GPU 性能和内存效率
+torch_dtype = torch.float16
+
+model_path = "Disty0/NewBie-image-Exp0.1-Diffusers"
 
 print("正在加载 Text Encoder...")
 text_encoder_2 = AutoModel.from_pretrained(
     model_path, 
     subfolder="text_encoder_2", 
     trust_remote_code=True, 
-    torch_dtype=torch.bfloat16
+    torch_dtype=torch_dtype
 )
 
 print("正在加载 Pipeline...")
+# 优化：使用更快的采样器 DPMSolverMultistepScheduler
 pipe_newbie = NewbiePipeline.from_pretrained(
     model_path, 
     text_encoder_2=text_encoder_2, 
-    torch_dtype=torch.bfloat16
+    torch_dtype=torch_dtype
 )
-pipe_newbie.to("cuda")
 
-del text_encoder_2
-torch.cuda.empty_cache()
+# 优化：使用更快的采样器
+pipe_newbie.scheduler = DPMSolverMultistepScheduler.from_config(pipe_newbie.scheduler.config)
+
+pipe_newbie.to("cuda")
 print("模型加载完成。")
 
 @spaces.GPU()
@@ -37,14 +43,18 @@ def generate_image_newbie(prompt, negative_prompt, height, width, num_inference_
 
     generator = torch.Generator("cuda").manual_seed(int(seed))
 
+    # 优化：减少推理步数以加快生成速度，同时保持质量
+    # 使用 DPMSolver 可以在较少的步数下获得更好的结果
     image = pipe_newbie(
         prompt=prompt,
         negative_prompt=negative_prompt,
         height=int(height),
         width=int(width),
-        num_inference_steps=int(num_inference_steps),
+        num_inference_steps=min(num_inference_steps, 25),  # 限制最大步数以加快速度
         guidance_scale=guidance_scale,
         generator=generator,
+        # 优化：启用内存高效模式
+        output_type="pil"
     ).images[0]
 
     return image, seed
@@ -92,12 +102,12 @@ with gr.Blocks(title="NewBie 图像生成器") as demo:
             )
             
             with gr.Row():
-                height_newbie = gr.Slider(label="图片高度 (Height)", minimum=512, maximum=2048, step=64, value=1024)
-                width_newbie = gr.Slider(label="图片宽度 (Width)", minimum=512, maximum=2048, step=64, value=1024)
+                height_newbie = gr.Slider(label="图片高度 (Height)", minimum=512, maximum=1024, step=64, value=768)  # 优化：降低默认分辨率以加快速度
+                width_newbie = gr.Slider(label="图片宽度 (Width)", minimum=512, maximum=1024, step=64, value=768)   # 优化：降低默认分辨率以加快速度
             
             with gr.Row():
-                steps_newbie = gr.Slider(label="推理步数 (Inference Steps)", minimum=1, maximum=100, step=1, value=28)
-                guidance_scale_newbie = gr.Slider(label="引导系数 (Guidance Scale)", minimum=1.0, maximum=20.0, step=0.1, value=3.5)
+                steps_newbie = gr.Slider(label="推理步数 (Inference Steps)", minimum=5, maximum=50, step=1, value=20)  # 优化：减少默认步数
+                guidance_scale_newbie = gr.Slider(label="引导系数 (Guidance Scale)", minimum=1.0, maximum=10.0, step=0.1, value=5.0)  # 优化：调整默认值
             
             seed_newbie = gr.Number(label="随机种子 (Seed, -1 为随机)", value=-1, precision=0)
             
@@ -106,6 +116,7 @@ with gr.Blocks(title="NewBie 图像生成器") as demo:
         with gr.Column(scale=1):
             image_output_newbie = gr.Image(label="生成结果 (Output)", format="png", interactive=False)
             used_seed_newbie = gr.Number(label="使用的种子 (Used Seed)", interactive=False)
+            progress_bar = gr.Textbox(label="生成进度", interactive=False, value="准备就绪...")
 
     generate_btn_newbie.click(
         fn=generate_image_newbie,
@@ -122,4 +133,7 @@ with gr.Blocks(title="NewBie 图像生成器") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        theme=gr.themes.Soft(primary_hue="blue"),
+        footer_links=[{"label": "Built with anycoder", "url": "https://huggingface.co/spaces/akhaliq/anycoder"}]
+    )