Skip to content

GitLab

Commit ead09e29 authored by Muyang Li's avatar Muyang Li
Browse files

chore: fix the ci version

parent 00ba7eaf
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
pyproject.toml
View file @ ead09e29
...@@ -32,18 +32,22 @@ dependencies = [ ...@@ -32,18 +32,22 @@ dependencies = [
"transformers>=4.53.3", "transformers>=4.53.3",
] ]
optional-dependencies.ci = [ optional-dependencies.ci = [
"accelerate==1.9",
"controlnet-aux==0.0.10", "controlnet-aux==0.0.10",
"datasets==3.6", "datasets==3.6",
"diffusers @ git+https://github.com/huggingface/diffusers@041501a", "diffusers @ git+https://github.com/huggingface/diffusers@041501a",
"facexlib==0.3", "facexlib==0.3",
"huggingface-hub==0.34",
"image-gen-aux @ git+https://github.com/asomoza/image_gen_aux.git", "image-gen-aux @ git+https://github.com/asomoza/image_gen_aux.git",
"insightface==0.7.3", "insightface==0.7.3",
"onnxruntime==1.22.1", "onnxruntime==1.22.1",
"opencv-python==4.11.0.86", "opencv-python==4.11.0.86",
"peft==0.17",
"pytest==8.4.2", "pytest==8.4.2",
"pytest-rerunfailures==16.0.1", "pytest-rerunfailures==16.0.1",
"timm==1.0.19", "timm==1.0.19",
"torchmetrics==1.8", "torchmetrics==1.8",
"transformers==4.53.3",
] ]
optional-dependencies.demo = [ optional-dependencies.demo = [
......
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