diff --git a/README.md b/README.md index c6fd697..13d6a18 100644 --- a/README.md +++ b/README.md @@ -46,7 +46,7 @@ pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f ht Install other required Python packages: ```sh -pip install ftfy regex tqdm omegaconf pytorch-lightning IPython kornia imageio imageio-ffmpeg einops torch_optimizer +pip install ftfy regex tqdm omegaconf pytorch-lightning IPython kornia imageio imageio-ffmpeg einops torch_optimizer==0.3.0 ``` Clone required repositories: