Upload folder using huggingface_hub

app.py

@@ -4,49 +4,49 @@ from typing import Optional
 
 import gradio as gr
 import torch
-from diffusers import AutoPipelineForText2Image
+from diffusers import AutoPipelineForText2Image, EulerAncestralDiscreteScheduler
 
 MODEL_ID = os.getenv("MODEL_ID", "stablediffusionapi/counterfeit-v30")
 DEFAULT_NEGATIVE = os.getenv(
     "DEFAULT_NEGATIVE",
-    "blurry, lowres, worst quality, low quality, bad anatomy, watermark, text, extra fingers",
+    "blurry, lowres, worst quality, low quality, bad anatomy, watermark, text, extra fingers, realistic, photo, 3d, child, loli, young-looking",
 )
 
 PRESETS = {
-    "🌻 向日葵少女": {
-        "prompt": "1girl, anime style, sunflower field, warm sunlight, soft lighting, detailed face, masterpiece, best quality",
+    "🎀 动漫头像": {
+        "prompt": "masterpiece, best quality, 1girl, solo, anime style, upper body portrait, detailed eyes, clean lineart, soft cel shading",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 18,
+        "steps": 12,
         "cfg": 7.0,
-        "width": 512,
-        "height": 768,
+        "width": 384,
+        "height": 576,
         "seed": -1,
     },
-    "🏙 赛博少女": {
-        "prompt": "1girl, anime style, cyberpunk city, neon lights, night street, glowing signs, dynamic pose, masterpiece",
+    "💼 成熟御姐": {
+        "prompt": "masterpiece, best quality, 1woman, mature female, solo, anime style, office lady, detailed eyes, upper body portrait, clean lineart",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 18,
+        "steps": 12,
         "cfg": 7.0,
-        "width": 512,
-        "height": 768,
+        "width": 384,
+        "height": 576,
         "seed": -1,
     },
     "🌸 樱花和服": {
-        "prompt": "1girl, anime style, kimono, cherry blossoms, spring, elegant pose, soft light, detailed illustration",
+        "prompt": "masterpiece, best quality, 1girl, anime style, kimono, cherry blossoms, elegant pose, detailed face, clean lineart",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 20,
+        "steps": 14,
         "cfg": 7.0,
-        "width": 512,
-        "height": 768,
+        "width": 448,
+        "height": 640,
         "seed": -1,
     },
-    "⚔️ 幻想剑士": {
-        "prompt": "anime warrior girl, fantasy armor, sword, dramatic lighting, wind, detailed background, masterpiece",
+    "🏙 赛博少女": {
+        "prompt": "masterpiece, best quality, 1girl, anime style, cyberpunk city, neon lights, upper body, detailed eyes, clean lineart",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 20,
+        "steps": 14,
         "cfg": 7.0,
-        "width": 512,
-        "height": 768,
+        "width": 448,
+        "height": 640,
         "seed": -1,
     },
 }
@@ -68,8 +68,12 @@ def load_pipe():
             pipe.safety_checker = None
         if hasattr(pipe, "requires_safety_checker"):
             pipe.requires_safety_checker = False
+        if hasattr(pipe, "scheduler") and pipe.scheduler is not None:
+            pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
         if hasattr(pipe, "enable_attention_slicing"):
             pipe.enable_attention_slicing()
+        if hasattr(pipe, "enable_vae_slicing"):
+            pipe.enable_vae_slicing()
         if hasattr(pipe, "vae") and hasattr(pipe.vae, "enable_tiling"):
             pipe.vae.enable_tiling()
         pipe = pipe.to(_device)
@@ -116,36 +120,37 @@ def generate(
     if seed >= 0:
         generator = torch.Generator(device=_device).manual_seed(int(seed))
 
-    image = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
-        num_inference_steps=int(steps),
-        guidance_scale=float(guidance_scale),
-        width=int(width),
-        height=int(height),
-        generator=generator,
-    ).images[0]
+    with torch.inference_mode():
+        image = pipe(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            num_inference_steps=int(steps),
+            guidance_scale=float(guidance_scale),
+            width=int(width),
+            height=int(height),
+            generator=generator,
+        ).images[0]
     return image
 
 
 with gr.Blocks(title="wode") as demo:
     gr.Markdown(
-        f"# wode\n\n免费档二次元生图（已切到 Counterfeit 路线）\n\n当前模型：`{MODEL_ID}`\n\n建议参数：`512x768`、`16-24 steps`、`CFG 6-8`。\n\n⚠️ 免费 CPU 很慢，首张图等 `1~3 分钟` 都正常。"
+        f"# wode\n\n免费档二次元生图（Counterfeit 路线）\n\n当前模型：`{MODEL_ID}`\n\n推荐先用：`384x576 / 12 steps / CFG 7`。\n\n想更像动漫角色：正向词尽量写 `1girl/1woman + anime style + detailed eyes + clean lineart`。\n想偏成年风格：加 `mature female` 或 `adult woman`。默认反向词已压制 `realistic/photo` 和 `child-like`。\n\n⚠️ 免费 CPU 仍然会慢，通常要等 `1~5 分钟`。"
     )
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
                 label="正向提示词",
                 lines=4,
-                placeholder="例如：1girl, anime style, detailed face, masterpiece, best quality",
+                placeholder="例如：masterpiece, best quality, 1girl, anime style, detailed eyes, clean lineart",
             )
-            negative_prompt = gr.Textbox(label="反向提示词", lines=2, value=DEFAULT_NEGATIVE)
+            negative_prompt = gr.Textbox(label="反向提示词", lines=3, value=DEFAULT_NEGATIVE)
             with gr.Row():
-                steps = gr.Slider(8, 32, value=20, step=1, label="步数 Steps")
+                steps = gr.Slider(6, 24, value=12, step=1, label="步数 Steps")
                 guidance_scale = gr.Slider(1, 12, value=7.0, step=0.5, label="引导强度 CFG")
             with gr.Row():
-                width = gr.Slider(256, 768, value=512, step=64, label="宽度")
-                height = gr.Slider(256, 1024, value=768, step=64, label="高度")
+                width = gr.Slider(256, 640, value=384, step=64, label="宽度")
+                height = gr.Slider(256, 896, value=576, step=64, label="高度")
             seed = gr.Number(label="随机种子（-1 为随机）", value=-1, precision=0)
             with gr.Row():
                 run = gr.Button("开始生成", variant="primary")
@@ -155,16 +160,16 @@ with gr.Blocks(title="wode") as demo:
 
     gr.Markdown("## 二次元预设按钮")
     with gr.Row():
-        preset1 = gr.Button("🌻 向日葵少女")
-        preset2 = gr.Button("🏙 赛博少女")
+        preset1 = gr.Button("🎀 动漫头像")
+        preset2 = gr.Button("💼 成熟御姐")
         preset3 = gr.Button("🌸 樱花和服")
-        preset4 = gr.Button("⚔️ 幻想剑士")
+        preset4 = gr.Button("🏙 赛博少女")
 
     for btn, name in [
-        (preset1, "🌻 向日葵少女"),
-        (preset2, "🏙 赛博少女"),
+        (preset1, "🎀 动漫头像"),
+        (preset2, "💼 成熟御姐"),
         (preset3, "🌸 樱花和服"),
-        (preset4, "⚔️ 幻想剑士"),
+        (preset4, "🏙 赛博少女"),
     ]:
         btn.click(
             fn=lambda n=name: apply_preset(n),
@@ -176,21 +181,21 @@ with gr.Blocks(title="wode") as demo:
         label="也可以直接点下面示例",
         examples=[
             [
-                PRESETS["🌻 向日葵少女"]["prompt"],
+                PRESETS["🎀 动漫头像"]["prompt"],
                 DEFAULT_NEGATIVE,
-                18,
+                12,
                 7.0,
-                512,
-                768,
+                384,
+                576,
                 -1,
             ],
             [
-                PRESETS["🏙 赛博少女"]["prompt"],
+                PRESETS["💼 成熟御姐"]["prompt"],
                 DEFAULT_NEGATIVE,
-                18,
+                12,
                 7.0,
-                512,
-                768,
+                384,
+                576,
                 -1,
             ],
         ],