Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ dependencies = [
"flash-linear-attention",
"timm",
"open-clip-torch>=3.2.0",
"mlflow>=3.2.0",
"mlflow>=3.5.0",
]


Expand All @@ -112,7 +112,8 @@ override-dependencies = [
"torchvision; sys_platform == 'never'",
"triton; sys_platform == 'never'",
"transformer-engine @ git+https://github.com/NVIDIA/TransformerEngine.git@6a34b6574fa6c29d9d07fdcddf9812cbb1488878",

"mlflow>=3.5.0",
"cryptography>=43.0.0,<47"
]

[tool.uv.sources]
Expand Down
Loading
Loading