CoolFace
Modelpublic

AXERA-TECH/lcm-lora-sdv1-5

sourceHugging Facebsd-3-clauseupdated 4mo agoView on Hugging Face
2likes23downloads
Model Card

SD1.5-LCM AX650N

本仓库是 lcm-lora-sdv1-5 的 AX650N 发布版本,包含 Stable Diffusion 1.5 LCM 文生图和图生图所需的 AX650N axmodel 文件。

AX620E 模型已经拆分到独立仓库:

text
AXERA-TECH/lcm-lora-sdv1-5-ax620e

发布内容只保留 *.axmodel 推理所需文件。*.onnx 和 *.safetensors 不属于当前发布路径。

原始模型请参考:

模型目录

OpenAI model id目录分辨率txt2imgimg2img
lcm-lora-sdv1-5-512x512models/512x512支持支持
lcm-lora-sdv1-5-768x1024models_1024x768/768x1024支持不支持

models_1024x768/ 当前没有 vae_encoder.axmodel,因此只支持文生图。

通过 axllm serve 启动

在 AX650N 板端执行:

sh
axllm serve /path/to/lcm-lora-sdv1-5 --port 8000

仓库也包含一个已经验证过的 AX650/AX650N aarch64 axllm 二进制,可直接在本仓库目录下运行:

sh
chmod +x ./bin/axllm
./bin/axllm serve . --port 8000

该二进制的来源和校验信息记录在 bin/axllm.version.json 中。

支持接口:

  • —GET /v1/models
  • —POST /v1/images/generations
  • —POST /v1/images/edits

前端或客户端应先请求 /v1/models 获取真实可用模型,再选择返回的模型 ID 发起生图请求。

文生图示例

sh
curl -X POST http://127.0.0.1:8000/v1/images/generations \
  -H 'Content-Type: application/json' \
  -d '{
    "model": "lcm-lora-sdv1-5-512x512",
    "prompt": "a cat wearing sunglasses",
    "size": "512x512",
    "seed": 0,
    "response_format": "b64_json"
  }'

图生图示例

sh
curl -X POST http://127.0.0.1:8000/v1/images/edits \
  -F model=lcm-lora-sdv1-5-512x512 \
  -F prompt="Astronauts in a jungle, cold color palette, muted colors, detailed, 8k" \
  -F size=512x512 \
  -F seed=0 \
  -F response_format=b64_json \
  -F image=@assets/img2img-init.png

Python Demo

sh
pip3 install -r requirements.txt
python3 launcher.py --model_dir models --isize 512x512
python3 launcher.py --model_dir models_1024x768 --isize 768x1024

性能参考

输入图片大小 512x512:

ModelsRaspberry Pi5 Only CPUIntel i7-13700Raspberry Pi5 + M.2 Card
UNet(1 step)14 s1.7 s0.43 s
VAE Encoder25 s1.7 s0.46 s
VAE Decoder58 s3.8 s0.91 s
Total txt2img, 4 steps120 s10.6 s2.68 s
Total img2img, 2 steps113 s8.9 s2.25 s