stabilityai/stable-diffusion-2-1
不是本機資料夾,而是 Hugging Face Model Hub 上的模型名稱。
當你第一次運行 Stable Diffusion 時,diffusers
會自動下載這個模型到你的 Hugging Face 缓存資料夾,通常是:
📍 Windows:
C:\Users\你的使用者名稱\.cache\huggingface\hub\
📍 Linux / Mac:
~/.cache/huggingface/hub/
✅ 如何解決?
方法 1️⃣:手動下載模型
你可以手動下載 Stable Diffusion 2.1 模型,然後指定本機路徑:
1️⃣ 打開 Hugging Face 網站
👉 https://huggingface.co/stabilityai/stable-diffusion-2-1
2️⃣ 下載模型檔案(stable-diffusion-2-1
)
- 下載
v2-1_512-ema-pruned.safetensors
或v2-1_768-ema-pruned.safetensors
- 存到你電腦的某個資料夾,例如:
C:\AI\Models\stable-diffusion-2-1\
3️⃣ 指定本機模型路徑 如果你不想從 Hugging Face 下載,可以手動指定模型路徑:
python train_network.py \
--pretrained_model_name_or_path="C:/AI/Models/stable-diffusion-2-1/" \
--train_data_dir="datasets/zenmaru_dataset" \
--output_dir="output/zenmaru_lora"
方法 2️⃣:確保 diffusers
能下載模型
如果你不希望手動下載,請確保你的 Python 環境 可以存取 Hugging Face:
✅ 安裝 diffusers
套件
pip install diffusers transformers accelerate
✅ 嘗試下載模型 你可以執行以下 Python 測試:
from diffusers import StableDiffusionPipeline
model_id = "stabilityai/stable-diffusion-2-1"
pipe = StableDiffusionPipeline.from_pretrained(model_id)
print("✅ 模型下載成功!")
如果報錯:
- 需要 Hugging Face API Token 👉 註冊 Hugging Face 帳號,並執行:
huggingface-cli login
🔥 總結
1️⃣ stabilityai/stable-diffusion-2-1
不是資料夾,而是 Hugging Face 模型名稱
2️⃣ 你可以手動下載模型,然後指定本機路徑
3️⃣ 如果想讓 Python 自動下載,請安裝 diffusers
,並確保 Hugging Face 能連線