[F] Fix deps

2025-04-08 22:47:00 +00:00
parent 5454087833
commit c19bc8683a
2 changed files with 1450 additions and 7 deletions
@@ -7,13 +7,22 @@ requires-python = ">=3.10"
 dependencies = [
    "accelerate>=1.2.1",
    "bitsandbytes>=0.45.5",
+    "datasets>=3.5.0",
+    "device-smi>=0.4.1",
    "diffusers>=0.32.1",
    "einops>=0.7.0",
    "flash-attn>=2.7.0",
    "gptqmodel>=2.2.0",
+    "gradio>=5.24.0",
+    "hf-transfer>=0.1.9",
+    "logbar>=0.0.4",
    "optimum>=1.24.0",
+    "pillow>=11.1.0",
+    "protobuf>=6.30.2",
    "sentencepiece>=0.2.0",
    "setuptools>=78.1.0",
+    "threadpoolctl>=3.6.0",
+    "tokenicer>=0.0.4",
    "torch>=2.5.1",
    "torchvision>=0.20.1",
    "transformers>=4.47.1",