From 39c3173b50b5e135564f826c1cded87f70a03fb4 Mon Sep 17 00:00:00 2001 From: Xinyu-Kang Date: Wed, 5 Nov 2025 15:49:18 +0000 Subject: [PATCH 1/3] added requirements-rocm.txt for AMD GPU and ROCm --- requirements-rocm.txt | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 requirements-rocm.txt diff --git a/requirements-rocm.txt b/requirements-rocm.txt new file mode 100644 index 000000000..c811fabdf --- /dev/null +++ b/requirements-rocm.txt @@ -0,0 +1,20 @@ +# Use the PyTorch ROCm wheel index (choose the stream that matches your system) +--extra-index-url https://download.pytorch.org/whl/rocm6.3 + +pre-commit +torch==2.8.0+rocm6.3 +torchaudio==2.8.0+rocm6.3 +torchvision==0.23.0+rocm6.3 +transformers==4.57.1 +qwen-vl-utils==0.0.11 +datasets +setuptools +tqdm +wandb +psutil +numpy +accelerate +pydantic +sglang[all]==0.5.4 +openai-harmony + From a6d781c12b412b7e984cbcb9e09ce17aeb4dd0ab Mon Sep 17 00:00:00 2001 From: Shenggui Li Date: Thu, 6 Nov 2025 15:28:09 +0800 Subject: [PATCH 2/3] fixed the workflow on CI runner (#276) * fixed the workflow on CI runner * polish * polish * polish * polish * polish * polish * polish * polish --- .github/workflows/test.yaml | 4 ++-- tests/test_target_modeling/test_auto_target_model.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 2823c851d..1286b6dce 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -29,6 +29,6 @@ jobs: pip install -e . - name: Run test - timeout-minutes: 10 + timeout-minutes: 30 run: | - python tests/ci/gpu_lock_exec.py --devices 6,7 python -m unittest discover -s ./tests -p "test_*.py" + python -m unittest discover -s ./tests -p "test_*.py" -v diff --git a/tests/test_target_modeling/test_auto_target_model.py b/tests/test_target_modeling/test_auto_target_model.py index bd9fc5e75..6bd3054e9 100644 --- a/tests/test_target_modeling/test_auto_target_model.py +++ b/tests/test_target_modeling/test_auto_target_model.py @@ -24,10 +24,10 @@ def test_auto_target_model(self): set_seed(42) hf_model = HFLlamaForCausalLM.from_pretrained( - "dphn/Dolphin3.0-Llama3.1-8B", torch_dtype=torch.bfloat16 + "unsloth/Llama-3.2-1B", torch_dtype=torch.bfloat16 ).cuda() sf_model = SFLlamaForCausalLM.from_pretrained( - "dphn/Dolphin3.0-Llama3.1-8B", torch_dtype=torch.bfloat16 + "unsloth/Llama-3.2-1B", torch_dtype=torch.bfloat16 ).cuda() # create data From 9637af61f493503ca11c9aa4f1d355c3ec7b5743 Mon Sep 17 00:00:00 2001 From: ChangLiu0709 Date: Thu, 6 Nov 2025 16:22:02 +0000 Subject: [PATCH 3/3] Added requirements-rocm.txt for the installation support of AMD GPUs and ROCm --- requirements-rocm.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/requirements-rocm.txt b/requirements-rocm.txt index c811fabdf..aac2dd331 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -17,4 +17,3 @@ accelerate pydantic sglang[all]==0.5.4 openai-harmony -