Gbone3176's picture
Fix Tr006 V2 online Segment preprocessing
a1408a1 verified
Raw History Blame Contribute Delete
1.06 kB
# Pins the torch already in the base image — the recommended CUDA 12.8 build (see
# the Dockerfile for why it is the default). Nothing is downloaded: pip treats the
# preinstalled build as satisfying this. Its only job is to turn "a dependency of
# mine wants a different torch" into an immediate, readable pip error naming that
# dependency, rather than a silent swap to the CUDA 13.0 wheel that cannot run on
# the L40S. Changing torch is allowed — if you do, edit this pin knowingly, and the
# Dockerfile's guard will tell you whether the result still suits the GPU you target.
torch==2.11.0
orena-focus>=0.3.4
# Direct imports of inference.py. They also arrive as transitive dependencies
# of orena-focus, but anything imported directly is declared explicitly.
decord>=0.6.0
numpy>=1.26.0
Pillow>=11.0.0
# Qwen3.5 support (including Qwen3_5ForConditionalGeneration and the
# timestamp-aware Qwen3VLProcessor) landed after the 4.x line. The merged
# checkpoint was produced and verified with this exact version.
transformers==5.15.0
accelerate>=1.10.1