Skip to content

Commit

Permalink
fix(pyproject): correct jetstream git revision
Browse files Browse the repository at this point in the history
  • Loading branch information
tengomucho committed Sep 10, 2024
1 parent c9f11da commit 525231a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ quality = ["black", "ruff", "isort"]
# Jetstream/Pytorch support is experimental for now, requires installation from fixed commit.
# Pallas is pulled because it will install a compatible version of jax[tpu].
jetstream-pt = [
"jetstream-pt @ git+https://github.com/google/jetstream-pytorch.git#df92015289953c506004e674d57651b03e4e89f2",
"jetstream-pt @ git+https://github.com/google/jetstream-pytorch.git@df92015289953c506004e674d57651b03e4e89f2",
"torch-xla[pallas] == 2.4.0"
]

Expand Down

0 comments on commit 525231a

Please sign in to comment.