DiffusionPipeline.from_pretrained 网络问题报错

Question

ldm = DiffusionPipeline.from_pretrained(model_id).to(device)
这个为什么始终报网络的错误呢

但是我明明开了代理，这个huggingface.com也可以正常ping通，而且前一天还是好好的😭😭

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

DiffusionPipeline.from_pretrained 网络问题报错

你遇到的问题是使用 DiffusionPipeline.from_pretrained 方法从 Hugging Face 的模型库中加载 Stable Diffusion 模型时，始终报网络错误。下面是一些可能的解决方案：

1. 检查模型 ID：确保你输入的 model_id 是正确的。可以尝试使用 huggingface.co/stable-diffusion-v1-4 这样的模型 ID。
2. 检查网络连接：确保你的网络连接是稳定的，并且可以访问 Hugging Face 的模型库。你可以尝试使用其他网络连接或代理服务器来检查是否是网络问题。
3. 检查模型大小：Stable Diffusion 模型非常大，可能会导致网络连接问题。你可以尝试使用 smaller 模型或分割模型来加载。
4. 检查 CUDA 设备：如果你使用了 CUDA 设备，请确保它是可用的。你可以尝试使用 nvidia-smi 命令来检查 CUDA 设备的状态。
5. 检查 PyTorch 版本：确保你使用的 PyTorch 版本是最新的。Stable Diffusion 模型可能需要最新的 PyTorch 版本来正确地加载。

以下是一个示例代码，使用 DiffusionPipeline.from_pretrained 方法加载 Stable Diffusion 模型：

import torch
from transformers import DiffusionPipeline

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model_id = "huggingface.co/stable-diffusion-v1-4"
ldm = DiffusionPipeline.from_pretrained(model_id).to(device)

如果你仍然遇到问题，可以提供更多的错误信息和代码，以便更好地帮助你解决问题。