ModuleNotFoundError: No module named 'ncsn'

[staguchi0703]

[/usr/local/lib/python3.11/dist-packages/py_img_gen/trainers/diffusion.py](https://localhost:8080/#) in <module>
      9 )
     10 from diffusers.utils import make_image_grid
---> 11 from ncsn.scheduler import AnnealedLangevinDynamicsScheduler as ALDScheduler
     12 from torch.optim import Optimizer
     13 from torch.utils.data import DataLoader

ModuleNotFoundError: No module named 'ncsn'

---

• ! pip install diffusers-ncsn

[shunk031]

@staguchi0703 コメントありがとうございます！ ご指摘の通り pip install diffusers-ncsn でインストールされます。
もしくは py-img-gen ライブラリをインストールする際に以下のようにしていただくといいかもしれません。

pip install py-img-gen[ncsn]