sd-0 / README.md
alfredplpl's picture
Update README.md
90f2fc7 verified
|
raw
history blame
858 Bytes
---
license: openrail++
---
# はじめに
このモデルはStable Diffusionを0から作りたい人のモデルです。どこから作るかは個々人にお任せします。
sdxl_0.safetensorsはU-Netを0に置き換えています。プログラムは次のとおりです。
```python
from safetensors import safe_open
from safetensors.torch import save_file
import torch
tensors = {}
with safe_open("/path/to/sd_xl_base_1.0.safetensors", framework="pt", device=0) as f:
for k in f.keys():
tensors[k] = f.get_tensor(k)
if("model.diffusion_model" in k):
tensors[k]=torch.zeros_like(tensors[k])
save_file(tensors, "/path/to/sdxl_0.safetensors")
```
このファイルを[sd-scripts](https://github.com/kohya-ss/sd-scripts)でファインチューンをしてください。学習率は2e-5がおすすめです。