当前位置：首页 » AI答疑

如何解决Step1X-Edit运行时的显存不足问题？

2025-08-23

1.5 K

解决显存不足的三种方案

针对Step1X-Edit运行时的显存需求（基础版需要42GB显存），以下是具体解决方案：

使用FP8量化版本：官方提供的FP8量化模型可将显存需求降至16GB，适合NVIDIA 3090 Ti等消费级显卡。安装时需在下载模型时选择step1x-edit-i1258-fp8.safetensors权重文件
调整分辨率参数：通过--size_level参数强制使用512×512分辨率（默认值），相比1024×1024可节省约8GB显存
分布式推理方案：对于专业研究场景，可使用官方提供的模型并行技术，通过修改scripts/dist_inference.py脚本将计算负载分配到多块GPU

额外建议：关闭不必要的后台程序释放显存，或在云服务平台租用A100/H800实例进行临时计算（约1.5美元/小时）