Skip to content

GitLab

Commit 927e75e1 authored by dcuai's avatar dcuai
Browse files

Update pyproject.toml ——bnb、deepspeed

parent 01947cc5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
pyproject.toml
View file @ 927e75e1
...@@ -17,14 +17,14 @@ dependencies = [ ...@@ -17,14 +17,14 @@ dependencies = [
"requests", "sentencepiece", "tokenizers>=0.12.1", "requests", "sentencepiece", "tokenizers>=0.12.1",
"torch", "torchvision", "wandb", "torch", "torchvision", "wandb",
"shortuuid", "httpx==0.24.0", "shortuuid", "httpx==0.24.0",
"deepspeed==0.12.3", #"deepspeed==0.12.3",
"peft==0.4.0", "peft==0.4.0",
"albumentations ", "albumentations ",
"opencv-python", "opencv-python",
"tiktoken", "tiktoken",
"accelerate==0.24.1", "accelerate==0.24.1",
"transformers==4.32.1", "transformers==4.32.1",
"bitsandbytes==0.41.0", # "bitsandbytes==0.41.0",
"scikit-learn==1.2.2", "scikit-learn==1.2.2",
"sentencepiece==0.1.99", "sentencepiece==0.1.99",
"einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13", "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment