timm for torch jit

requirements.txt

@@ -1,24 +1,13 @@
-accelerate==0.24.1
-decorator==4.4.2
-decord==0.6.0
-gradio==3.47.1
-gradio_client==0.6.0
-huggingface-hub==0.17.3
 iopath==0.1.10
-moviepy==1.0.3
 omegaconf==2.3.0
 opencv-python==4.7.0.72
 peft==0.2.0
 soundfile==0.12.1
-spacy==3.7.4
-spacy-legacy==3.0.12
-spacy-loggers==1.0.5
-timm==0.6.13
-tokenizers==0.14.1
-torch==2.0.0
-torchaudio==2.0.1
-torchvision==0.15.1
-tqdm==4.66.1
-transformers==4.35.0
 wandb==0.16.0
-webdataset==0.2.48
+webdataset==0.2.48
+torchvision
+transformers
+tqdm
+timm
+gradio==3.47.1
+gradio_client==0.6.0