Stable Diffusion本地部署踩坑记

之前踩过这个坑，花了不少时间才解决。把关键步骤整理出来，核心代码都贴上了，直接拿去用就行。

先搞清楚原理

要理解Stable Diffusion，先得搞清楚模型参数优化这个机制。直接看代码比看文字描述来得快：

from transformers import pipeline, AutoTokenizer

task = ["sentiment-analysis", "text-classification", "zero-shot-classification"][2]
model_name = "bert-base-chinese"

tokenizer = AutoTokenizer.from_pretrained(model_name)
print(f"词表大小: {tokenizer.vocab_size}")

nlp = pipeline(task, model=model_name)

texts = [
    "这个产品非常好用，值得推荐",
    "服务态度很差，不推荐",
    "质量一般，性价比不高",
]

results = nlp(texts[:3])
for text, result in zip(texts, results):
    print(f"{text[:20]}... -> {result}")

实际怎么用

实际编码中，Stable Diffusion的用法可以简化。下面是我在项目中用的写法，经过几轮迭代优化过：

import asyncio
from openai import AsyncOpenAI

client = AsyncOpenAI()

async def process_batch(prompts, concurrency=5):
    semaphore = asyncio.Semaphore(concurrency)
    async def call_api(prompt):
        async with semaphore:
            resp = await client.chat.completions.create(
                model="gpt-4", messages=[{"role": "user", "content": prompt}], max_tokens=200,
            )
            return resp.choices[0].message.content
    tasks = [call_api(p) for p in prompts]
    return await asyncio.gather(*tasks)

prompts = [f"用一句话解释第{j}个概念" for j in range(20)]
results = asyncio.run(process_batch(prompts))
for j, r in enumerate(results):
    print(f"[{j}] {r[:60]}")

进阶一点的用法

进阶一点的话，Stable Diffusion还有模型参数优化这个方向可以探索。原理不展开，直接看关键代码：

容易踩的坑

踩坑经验分享：Stable Diffusion在模型参数优化方面有几个暗坑。最常见的是参数默认值不符合预期，其次是并发场景下的竞态条件。建议加上防御性检查。

写在最后

Stable Diffusion这块先整理到这里。模型评估是最重要的部分，务必理解透彻。如果遇到问题，先检查配置和版本，大部分报错都是这两个原因。