Update pyproject.toml ——bnb、deepspeed

927e75e1 · dcuai · 01947cc5 · 927e75e1
Commit 927e75e1 authored Oct 25, 2024 by dcuai
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

pyproject.toml pyproject.toml +2 -2

No files found.
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,14 +17,14 @@ dependencies = [
    "requests", "sentencepiece", "tokenizers>=0.12.1",
    "torch", "torchvision", "wandb",
    "shortuuid", "httpx==0.24.0",
-    "deepspeed==0.12.3",
+    #"deepspeed==0.12.3",
    "peft==0.4.0",
    "albumentations ",
    "opencv-python",
    "tiktoken",
    "accelerate==0.24.1",
    "transformers==4.32.1",
-    "bitsandbytes==0.41.0",
+   # "bitsandbytes==0.41.0",
    "scikit-learn==1.2.2",
    "sentencepiece==0.1.99",
    "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",