diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index fe5859e5..d1f62931 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -8,6 +8,7 @@ on: paths: - "**/*.py" - "requirements.txt" + - "Makefile" - ".github/workflows/*.yml" pull_request: branches: @@ -15,6 +16,7 @@ on: paths: - "**/*.py" - "requirements.txt" + - "Makefile" - ".github/workflows/*.yml" jobs: diff --git a/Makefile b/Makefile index d4ca38e2..fda05fa7 100644 --- a/Makefile +++ b/Makefile @@ -21,4 +21,4 @@ style: ruff format $(check_dirs) test: - CUDA_VISIBLE_DEVICES= WANDB_DISABLED=true pytest -vv tests/ + CUDA_VISIBLE_DEVICES= ASCEND_RT_VISIBLE_DEVICES=0 WANDB_DISABLED=true pytest -vv --import-mode=importlib tests/ tests_v1/ diff --git a/src/llamafactory/v1/extras/__init__.py b/src/llamafactory/v1/extras/__init__.py new file mode 100644 index 00000000..8b137891 --- /dev/null +++ b/src/llamafactory/v1/extras/__init__.py @@ -0,0 +1 @@ +