Fix dependencies definition

dora-rs · Jan 17, 2025 · c361020 · c361020
1 parent c04a73b
commit c361020
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/node-hub/dora-qwenvl/pyproject.toml b/node-hub/dora-qwenvl/pyproject.toml
@@ -20,7 +20,7 @@ dependencies = [
     "accelerate >= 0.33",
     "opencv-python >= 4.1.1",
     "modelscope >= 1.18.1",
-    "peft = 0.13.2",
+    "peft == 0.13.2",
 ]
 
 # flash_attn = "^2.6.1" # Install using: pip install -U flash-attn --no-build-isolation

diff --git a/node-hub/dora-rdt-1b/pyproject.toml b/node-hub/dora-rdt-1b/pyproject.toml
@@ -17,14 +17,14 @@ dependencies = [
     "accelerate >= 0.33",
     "opencv-python >= 4.1.1",
     "modelscope >= 1.18.1",
-    "packaging = 24.0",
-    "wandb =0.17.0",
-    "diffusers =0.27.2",
-    "timm =1.0.3",
-    "sentencepiece =0.2.0",
-    "h5py =3.11.0",
-    "imgaug =0.4.0",
-    "huggingface_hub = 0.23.5",
+    "packaging == 24.0",
+    "wandb == 0.17.0",
+    "diffusers == 0.27.2",
+    "timm == 1.0.3",
+    "sentencepiece == 0.2.0",
+    "h5py == 3.11.0",
+    "imgaug == 0.4.0",
+    "huggingface_hub == 0.23.5",
 ]
 
 # flash_attn = "^2.6.1" # Install using: pip install -U flash-attn --no-build-isolation