Skip to content
Open
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 dataflow/run-inference/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
apache-beam[gcp]==2.49.0
torch==2.2.2
torch==2.8.0
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

critical

Updating torch to 2.8.0 while keeping apache-beam at 2.49.0 is likely to cause a critical dependency conflict that will prevent the Dataflow pipeline from running. apache-beam 2.49.0 (released in 2023) has strict dependencies, such as numpy<1.25.0, which are incompatible with the requirements of torch 2.8.0 (released in 2025). This conflict will likely cause the environment build to fail on Dataflow workers.

Furthermore, to fully address the security concerns mentioned in this PR (such as CVE-2025-32434), it is essential to use weights_only=True when loading models. This feature is supported in newer versions of apache-beam (e.g., 2.60.0+) via the torch_load_params argument in PytorchModelHandlerTensor, but is not available in version 2.49.0.

Please consider updating apache-beam to a more recent version to ensure compatibility and to enable the use of critical security features.

transformers==4.38.0