Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "af5551f929b1906345abd5866959c12c83c7bdbd"
Commit 7a2d539b authored by chenpangpang's avatar chenpangpang
Browse files

feat: modify requirements for dcu

parent f8744c3a
Pipeline #1355 failed with stages
in 0 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 15 deletions
Kolors/requirements.txt
View file @ 7a2d539b
fire
triton
pydantic
accelerate==0.27.2
deepspeed==0.8.1 deepspeed==0.8.1
huggingface-hub==0.20.2
imageio==2.25.1 imageio==2.25.1
numpy==1.21.6
omegaconf==2.3.0 omegaconf==2.3.0
pandas==1.3.5
Pillow==9.4.0
tokenizers==0.13.2
torch==1.13.1
torchvision==0.14.1
transformers==4.26.1
xformers==0.0.16
safetensors==0.3.3
diffusers==0.28.2 diffusers==0.28.2
sentencepiece==0.1.99
gradio==3.38.0 gradio==3.38.0
matplotlib-inline matplotlib-inline
ipython ipython
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