guided-diffusion を Google Colaboratory で実装したい

前提・ 実現したいこと

下記の Diffusion model を Google Colaboratory で試しています。(初学者です)
https://github.com/openai/guided-diffusion

README に従い models フォルダの作成、そこに学習済みモデルのチェックポイントのダウンロードまで終了しています。

発生している問題・エラーメッセージ

学習済みモデルからのサンプリングを行いたいのですが、コードを実行すると下記のエラーメッセージが出ます。

python

from guided_diffusion import dist_util, logger ModuleNotFoundError: No module named 'guided_diffusion'

該当のソースコード

python

!git clone https://github.com/openai/guided-diffusion.git # 学習済みモデルのダウンロード%cd /content/guided-diffusion/models !wget https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_diffusion.pt !wget https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_diffusion_uncond.pt !wget https://openaipublic.blob.core.windows.net/diffusion/jul-2021/256x256_classifier.pt %cd .. !pip install . %cd /content/guided-diffusion/scripts # 学習済みモデルからのサンプリング!python classifier_sample.py !python image_sample.py !python super_res_sample.py SAMPLE_FLAGS="--batch_size 4 --num_samples 100 --timestep_respacing 250"

補足情報（FW/ツールのバージョンなど）