Set ranges for deps

Signed-off-by: Thara Palanivel <[email protected]>

Author: tharapalanivel

pyproject.toml

@@ -21,25 +21,26 @@ classifiers=[
 ]
 dynamic = ["version"]
 dependencies = [
-"numpy",
-"tqdm",
+"numpy>=1.26.4,<2.0",
+"accelerate>=0.20.3,!=0.34,<1.1",
+"transformers>=4.45,<4.46",
+"torch>=2.2.0,<2.4", 
+"tqdm>=4.66.2,<5.0",
+"datasets>=2.15.0,<3.0",
+"ninja>=1.11.1.1,<2.0",
 "tensorboard",
 "notebook",
-"transformers",
-"accelerate",
+"torchvision>=0.8",
 "evaluate",
-"datasets",
 "huggingface_hub",
 "pandas",
 "safetensors",
-"ninja",
-"accelerate>=0.20.3,!=0.34,<1.1",
 ]
 
 [project.optional-dependencies]
 dev = ["pre-commit>=3.0.4,<5.0"]
 fp8 = ["llmcompressor"]
-gptq = ["auto_gptq"]
+gptq = ["auto_gptq>0.4.2", "optimum>=1.15.0"]
 visualize = ["matplotlib", "graphviz", "pygraphviz"]
 flash-attn = ["flash-attn>=2.5.3,<3.0"]
 opt = ["fms-model-optimizer[fp8, gptq]"]