|
|
|
@@ -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" |