提示显存不足
RuntimeError: CUDA out of memory. Tried to allocate 1.50 GiB (GPU 0; 8.00 GiB total capacity; 5.62 GiB already allocated; 109.75 MiB free; 5.74 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
1、添加参数–n_samples 1，其默认值为3
python scripts/txt2img.py --prompt “a photograph of an astronaut riding a horse” --plms –n_samples 1
文章来源：https://www.toymoban.com/news/detail-781390.html
2、降低精度
修改scripts/txt2img.py文件，在model.to(device)后添加.half()
文章来源地址https://www.toymoban.com/news/detail-781390.html
3、结果<
到了这里，关于AI绘画——使用stable-diffusion生成图片时提示RuntimeError: CUDA out of memory处理方法的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
