damccorm commented on code in PR #36420:
URL: https://github.com/apache/beam/pull/36420#discussion_r2478823627
##########
sdks/python/container/run_generate_requirements.sh:
##########
@@ -76,14 +76,24 @@ python"${PY_VERSION}" -m venv "$ENV_PATH"
source "$ENV_PATH"/bin/activate
pip install --upgrade pip setuptools wheel
+# For non-vllm (non-gpu) requirement files, force downloading torch from CPU
wheels
Review Comment:
Yes - specifically
https://github.com/apache/beam/blob/fe07fe767098663c032f84df74b4b4f7723bf607/sdks/python/container/ml/py310/ml_image_requirements.txt#L217
has a cpu postfix, and some gpu requirements (which are not ASF-license
compliant) are omitted
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
