[docker] patch docker-rocm (#7725)

* Update Dockerfile

* Fix typo

* Fix syntax for /bin/sh conditional

* Add build args to docker-compose

* Change shell to /bin/bash

This is required for "==" syntax in conditional string comparison
This commit is contained in:
Joe Schoonover
2025-04-15 01:36:39 -04:00
committed by GitHub
parent ac8c6fdd3a
commit b5d667cebf
2 changed files with 14 additions and 0 deletions

View File

@@ -9,8 +9,10 @@ services:
INSTALL_DEEPSPEED: "false"
INSTALL_FLASHATTN: "false"
INSTALL_LIGER_KERNEL: "false"
INSTALL_PYTORCH: "true"
INSTALL_HQQ: "false"
PIP_INDEX: https://pypi.org/simple
PYTORCH_INDEX: https://download.pytorch.org/whl/nightly/rocm6.3
container_name: llamafactory
volumes:
- ../../hf_cache:/root/.cache/huggingface