Upload folder using huggingface_hub

README.md

@@ -11,7 +11,7 @@ license: mit
 
 # wode
 
-Free lightweight image generation Space.
+Free anime image generation Space.
 
-Default model: `stabilityai/sd-turbo`
-Recommended settings on free CPU: 512x512, 1-4 steps.
+Default model: `stablediffusionapi/counterfeit-v30`
+Recommended settings on free CPU: 512x768, 16-24 steps, expect slow generations.

app.py

@@ -6,36 +6,36 @@ import gradio as gr
 import torch
 from diffusers import AutoPipelineForText2Image
 
-MODEL_ID = os.getenv("MODEL_ID", "stabilityai/sd-turbo")
+MODEL_ID = os.getenv("MODEL_ID", "stablediffusionapi/counterfeit-v30")
 DEFAULT_NEGATIVE = os.getenv(
     "DEFAULT_NEGATIVE",
-    "blurry, lowres, worst quality, low quality, bad anatomy",
+    "blurry, lowres, worst quality, low quality, bad anatomy, watermark, text, extra fingers",
 )
 
 PRESETS = {
     "🌻 向日葵少女": {
         "prompt": "1girl, anime style, sunflower field, warm sunlight, soft lighting, detailed face, masterpiece, best quality",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 2,
-        "cfg": 0.0,
+        "steps": 18,
+        "cfg": 7.0,
         "width": 512,
-        "height": 512,
+        "height": 768,
         "seed": -1,
     },
     "🏙 赛博少女": {
         "prompt": "1girl, anime style, cyberpunk city, neon lights, night street, glowing signs, dynamic pose, masterpiece",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 2,
-        "cfg": 0.0,
+        "steps": 18,
+        "cfg": 7.0,
         "width": 512,
-        "height": 512,
+        "height": 768,
         "seed": -1,
     },
     "🌸 樱花和服": {
         "prompt": "1girl, anime style, kimono, cherry blossoms, spring, elegant pose, soft light, detailed illustration",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 2,
-        "cfg": 0.0,
+        "steps": 20,
+        "cfg": 7.0,
         "width": 512,
         "height": 768,
         "seed": -1,
@@ -43,8 +43,8 @@ PRESETS = {
     "⚔️ 幻想剑士": {
         "prompt": "anime warrior girl, fantasy armor, sword, dramatic lighting, wind, detailed background, masterpiece",
         "negative": DEFAULT_NEGATIVE,
-        "steps": 2,
-        "cfg": 0.0,
+        "steps": 20,
+        "cfg": 7.0,
         "width": 512,
         "height": 768,
         "seed": -1,
@@ -126,22 +126,22 @@ def generate(
 
 with gr.Blocks(title="wode") as demo:
     gr.Markdown(
-        f"# wode\n\n免费轻量版二次元生图\n\n当前模型：`{MODEL_ID}`\n\n建议参数：`512x512`、`1-4 steps`。免费档先求快，再求细。"
+        f"# wode\n\n免费档二次元生图（已切到 Counterfeit 路线）\n\n当前模型：`{MODEL_ID}`\n\n建议参数：`512x768`、`16-24 steps`、`CFG 6-8`。\n\n⚠️ 免费 CPU 很慢，首张图等 `1~3 分钟` 都正常。"
     )
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
                 label="正向提示词",
                 lines=4,
-                placeholder="例如：1girl, anime style, sunflower field, warm sunlight, masterpiece",
+                placeholder="例如：1girl, anime style, detailed face, masterpiece, best quality",
             )
             negative_prompt = gr.Textbox(label="反向提示词", lines=2, value=DEFAULT_NEGATIVE)
             with gr.Row():
-                steps = gr.Slider(1, 8, value=2, step=1, label="步数 Steps")
-                guidance_scale = gr.Slider(0, 5, value=0.0, step=0.5, label="引导强度 CFG")
+                steps = gr.Slider(8, 32, value=20, 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, 768, value=512, step=64, label="高度")
+                height = gr.Slider(256, 1024, value=768, step=64, label="高度")
             seed = gr.Number(label="随机种子（-1 为随机）", value=-1, precision=0)
             with gr.Row():
                 run = gr.Button("开始生成", variant="primary")
@@ -174,19 +174,19 @@ with gr.Blocks(title="wode") as demo:
             [
                 PRESETS["🌻 向日葵少女"]["prompt"],
                 DEFAULT_NEGATIVE,
-                2,
-                0.0,
-                512,
+                18,
+                7.0,
                 512,
+                768,
                 -1,
             ],
             [
                 PRESETS["🏙 赛博少女"]["prompt"],
                 DEFAULT_NEGATIVE,
-                2,
-                0.0,
-                512,
+                18,
+                7.0,
                 512,
+                768,
                 -1,
             ],
         ],