Merge pull request #118 from Clarifai/model-examples-requirements

[EAGLE-5499] Standardized the requirements.txt files for all the examples

models/model_upload/image-classifier/nsfw-image-classifier/requirements.txt

@@ -1,8 +1,9 @@
-torch==2.4.1
-tokenizers==0.19.1
-transformers==4.44.1
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
 pillow==10.4.0
 requests==2.32.3
+timm==1.0.12
 opencv-python-headless==4.10.0.84
-aiohttp
 numpy
+aiohttp

models/model_upload/image-detector/detr-resnet-image-detection/requirements.txt

@@ -1,6 +1,6 @@
-torch==2.4.1
-tokenizers==0.19.1
-transformers==4.44.2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
 pillow==10.4.0
 requests==2.32.3
 timm==1.0.12

models/model_upload/llms/llama-3-8b-instruct/requirements.txt

@@ -1,13 +1,11 @@
-torch==2.3.1
-tokenizers==0.19.1
-transformers==4.44.2
-accelerate==0.34.2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
 scipy==1.10.1
-optimum==1.22.0
-xformers==0.0.27
+optimum>=1.23.3
+xformers==0.0.28.post3
 protobuf==5.27.3
-einops==0.8.0
+einops>=0.8.0
 requests==2.32.2
-sentence_transformers==2.2.0
-sentencepiece==0.2.0
-autoawq==0.2.6
+autoawq==0.2.7.post3

models/model_upload/llms/llama-3_2-1b-instruct/requirements.txt

@@ -1,12 +1,10 @@
-torch==2.4.0
-tokenizers==0.19.0
-transformers==4.44.2
-accelerate==0.34.2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
 scipy==1.10.1
-optimum==1.22.0
-xformers==0.0.27.post2
+optimum>=1.23.3
+xformers==0.0.28.post3
 protobuf==5.27.3
-einops==0.8.0
+einops>=0.8.0
 requests==2.32.2
-sentence_transformers==2.2.0
-sentencepiece==0.2.0

models/model_upload/llms/lmdeploy-llama-3_2-1b-instruct/requirements.txt

@@ -1,15 +1,14 @@
-torch==2.4.0
-tokenizers==0.20.2
-transformers==4.46.2
-accelerate==0.34.2
-optimum==1.23.3
-xformers==0.0.27.post2
-einops==0.8.0
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
+scipy==1.10.1
+optimum>=1.23.3
+xformers==0.0.28.post3
+protobuf==5.27.3
+einops>=0.8.0
 requests==2.32.2
-packaging
-ninja
-protobuf==3.20.0
 
-openai==1.51.0
+openai==1.61.0
 qwen-vl-utils
-lmdeploy==0.6.3
+lmdeploy>=0.6.4

models/model_upload/llms/vllm-mistral-7b-instruct/requirements.txt

@@ -1,17 +1,14 @@
-torch==2.3.1
-tokenizers==0.19.1
-transformers==4.44.2
-accelerate==0.34.2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
 scipy==1.10.1
-optimum==1.22.0
-xformers==0.0.27
+optimum>=1.23.3
+xformers==0.0.28.post3
 protobuf==5.27.3
-einops==0.8.0
+einops>=0.8.0
 requests==2.32.2
-sentence_transformers==2.2.0
-sentencepiece==0.2.0
-packaging
-ninja
-openai==1.51.0
-vllm==0.5.3
-autoawq==0.2.6
+openai==1.61.0
+vllm==0.7.1
+autoawq==0.2.7.post3
+

models/model_upload/multimodal_models/vllm-miniCPM-2.6/requirements.txt

@@ -1,18 +1,15 @@
-torch==2.4.0
-tokenizers==0.19.1
-transformers==4.44.2
-accelerate==0.34.2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
 scipy==1.10.1
-optimum==1.22.0
-xformers==0.0.27.post2
-einops==0.8.0
+optimum>=1.23.3
+xformers==0.0.28.post3
+protobuf==5.27.3
+einops>=0.8.0
+requests==2.32.2
+openai==1.61.0
+vllm==0.7.1
+autoawq==0.2.7.post3
 pillow==10.4.0
-requests==2.32.3
-sentence_transformers==2.2.0
-sentencepiece==0.2.0
-openai==1.55.3
-vllm==0.6.0
-autoawq==0.2.7.post2
 decord
-packaging
-ninja

models/model_upload/ocr/got-ocr2.0/requirements.txt

@@ -1,15 +1,13 @@
-torch==2.4.0
-tokenizers==0.19.0
-transformers==4.44.2
-accelerate==0.34.2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
 scipy==1.10.1
-optimum==1.22.0
-xformers==0.0.27.post2
+optimum>=1.23.3
+xformers==0.0.28.post3
 protobuf==5.27.3
-einops==0.8.0
+einops>=0.8.0
 requests==2.32.2
-sentence_transformers==2.2.0
-sentencepiece==0.2.0
-torchvision==0.19.0
 verovio==4.3.1
 tiktoken==0.8.0
+torchvision

models/model_upload/test-upload/mbart/requirements.txt

@@ -1,8 +1,8 @@
-torch==2.4.0
-tokenizers==0.21.0
-transformers>=4.47
-accelerate==1.2.0
-optimum==1.23.3
-xformers==0.0.27.post2
+torch==2.5.1
+tokenizers>=0.21.0
+transformers>=4.47.0
+accelerate>=1.2.0
+optimum>=1.23.3
+xformers==0.0.28.post3
 sentencepiece==0.2.0
-requests==2.23.0
+requests==2.32.2