fixed the workflow on CI runner (#276)

* fixed the workflow on CI runner

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

Author: FrankLeeeee

.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

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