diff --git a/settings.ini b/settings.ini index 153a79a..f27bfe0 100755 --- a/settings.ini +++ b/settings.ini @@ -34,7 +34,7 @@ clean_ids = True clear_all = False cell_number = True requirements = matplotlib datasets fastprogress fastcore -pip_requirements = torch torcheval diffusers einops timm wandb mlflow accelerate tensorboard +pip_requirements = torch torcheval diffusers einops timm wandb mlflow accelerate tensorboard torchvision torchaudio conda_requirements = pytorch>=1.7,<1.14 dev_requirements = nbdev