Skip to content
Closed
Changes from all 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
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -715,7 +715,7 @@ def _read_requirements(filename: str) -> list[str]:
], # Required for audio processing
"video": [], # Kept for backwards compatibility
# FlashInfer should be updated together with the Dockerfile
"flashinfer": ["flashinfer-python==0.3.1"],
"flashinfer": ["flashinfer-python==0.4.0rc4"],
# Optional deps for AMD FP4 quantization support
Comment on lines 715 to 719

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

P1 Badge Keep Docker flashinfer version in sync

The change bumps the flashinfer extra to 0.4.0rc2, but the repo’s Dockerfile still installs flashinfer_python-0.3.1. The comment above this stanza explicitly says FlashInfer should be updated together with the Dockerfile, so Docker builds will continue to use the old wheel and miss the determinism fix this commit is targeting. Consider updating the pinned version in docker/Dockerfile to avoid version skew between local installs and the official image.

Useful? React with 👍 / 👎.

"petit-kernel": ["petit-kernel"],
},
Expand Down