Skip to content

Commit

Permalink
Use poetry for rdt 1b as it seems to fix a submodule path issue
Browse files Browse the repository at this point in the history
  • Loading branch information
haixuanTao committed Jan 17, 2025
1 parent 5205102 commit 868f1f3
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions node-hub/dora-rdt-1b/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,11 @@ authors = [{ name = "Haixuan Xavier Tao", email = "[email protected]" }]
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",
Expand Down Expand Up @@ -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"

0 comments on commit 868f1f3

Please sign in to comment.