more bearable pyproject.toml

pyproject.toml

@@ -1,5 +1,6 @@
 [build-system]
-requires = ["setuptools>=69.5"]
+# requires = ["setuptools>=69.5"]
+requires = ["setuptools"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools.packages.find]
@@ -13,20 +14,23 @@ readme = "README.md"
 requires-python = ">=3.10"
 classifiers = ["Programming Language :: Python :: 3"]
 dependencies = [
-    "accelerate==1.1.0",
+    # "accelerate==1.1.0",
+    "accelerate",
     "loguru",
     "requests",
-    "setuptools==69.5",
+    # "setuptools>=69.5",
+    "setuptools",
     "transformers",
     "tokenizers",
-    "flash-attn==2.5.8",
-    "bitsandbytes==0.43.1",
+    # "flash-attn==2.5.8",
+    "flash-attn",
+    # "bitsandbytes==0.43.1",
+    "bitsandbytes",
     "safetensors",
     "gpustat",
     "icecream",
     "pdf2image",
-    "numpy==1.26.4",
-    "torchvision",
+    "numpy",
     "jsonlines",
     "editdistance",
     "einops",
@@ -34,13 +38,17 @@ dependencies = [
     "peft",
     "timm",
     "sentencepiece",
-    "colpali-engine==0.3.1",
+    # "colpali-engine==0.3.1",
+    "colpali-engine",
     "easyocr",
     "qwen-vl-utils",
     "faiss-cpu",
     "word2number",
-    "datasets>=3.0.0",
+    # "datasets>=3.0.0",
     "python-dotenv",
+    # "torchvision==0.22.1+cu128",
+    "torchvision",
+    "datasets>=3.6.0",
 ]
 
 [tool.ruff]