mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-23 01:29:48 +00:00
5242d986f4
*Description of changes:* This PR removes casting to `fp32` for the `cumsum` operation and upgrades `mlx` to `~=0.10.0` which adds `bf16` support for `cumsum`. Related: https://github.com/ml-explore/mlx/issues/959 By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. Co-authored-by: Abdul Fatir Ansari <ansarnd@amazon.com>
18 lines
381 B
TOML
18 lines
381 B
TOML
[project]
|
|
name = "chronos_mlx"
|
|
version = "1.1.0"
|
|
requires-python = ">=3.8"
|
|
license = { file = "LICENSE" }
|
|
dependencies = [
|
|
"torch~=2.1", # package was tested on 2.2
|
|
"transformers~=4.31",
|
|
"accelerate",
|
|
"mlx~=0.10.0"
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
test = ["pytest~=8.0", "numpy~=1.21"]
|
|
typecheck = ["mypy~=1.9"]
|
|
|
|
[tool.mypy]
|
|
ignore_missing_imports = true
|