Update settings.ini

m-gopichand · Dec 26, 2024 · 8950c35 · 8950c35
1 parent 40e80a1
commit 8950c35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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