Browse Source

Use poetry for rdt 1b as it seems to fix a submodule path issue

tags/v0.3.9-rc1
haixuantao 1 year ago
parent
commit
868f1f31fe
1 changed files with 6 additions and 2 deletions
  1. +6
    -2
      node-hub/dora-rdt-1b/pyproject.toml

+ 6
- 2
node-hub/dora-rdt-1b/pyproject.toml View File

@@ -5,11 +5,11 @@ authors = [{ name = "Haixuan Xavier Tao", email = "tao.xavier@outlook.com" }]
description = "Dora Node for RDT 1B"
license = { text = "MIT" }
readme = "README.md"
requires-python = ">=3.8"
requires-python = ">=3.9"

dependencies = [
"dora-rs >= 0.3.6",
"numpy < 2.0.0",
"numpy > 1.24.4",
"torch >= 2.4.0",
"torchvision >= 0.19",
"transformers >= 4.45",
@@ -43,3 +43,7 @@ dora-rdt-1b = "dora_rdt_1b.main:main"

[tool.ruff]
exclude = ["dora_rdt_1b/RoboticsDiffusionTransformer"]

[build-system]
requires = ["poetry-core>=1.8.0"]
build-backend = "poetry.core.masonry.api"

Loading…
Cancel
Save