Skip to content

Commit 68488d5

Browse files
authored
chore: update the test ci for the new comfyui tests (#750)
* chore: update the comfyui test ci * debugging * update * update * udpate" * use conda's c++ * update * update
1 parent f05237e commit 68488d5

2 files changed

Lines changed: 22 additions & 15 deletions

File tree

.github/workflows/pr-test.yaml

Lines changed: 21 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ jobs:
4040
uses: actions/checkout@v4
4141
with:
4242
submodules: true
43+
ref: ${{ github.event_name == 'workflow_dispatch' && github.event.inputs.ref || '' }}
4344
- name: Show current commit
4445
run: git log -1 --oneline
4546
- name: Set up Python
@@ -53,40 +54,46 @@ jobs:
5354
conda activate test_env || { echo "Failed to activate conda env"; exit 1; }
5455
which python
5556
echo "Installing dependencies"
57+
conda install -c conda-forge gxx=11 gcc=11 libsndfile -y
5658
pip install torch==2.8 torchvision==0.23 --index-url https://download.pytorch.org/whl/cu128
5759
pip install -e ".[ci]"
5860
- name: Setup ComfyUI
5961
run: |
6062
source $(conda info --base)/etc/profile.d/conda.sh
6163
conda activate test_env || { echo "Failed to activate conda env"; exit 1; }
6264
which python
63-
pwd
6465
cd ..
65-
pip install comfy-cli
6666
rm -rf ComfyUI
67-
yes | comfy --here install --nvidia --skip-torch-or-directml --version 0.3.44
67+
mkdir -p ComfyUI
6868
cd ComfyUI
69-
rm -r models
70-
mkdir -p $COMFYUI_MODELS_ROOT
71-
ln -s $COMFYUI_MODELS_ROOT models
69+
mkdir -p "${COMFYUI_MODELS_ROOT}"
70+
ln -s "${COMFYUI_MODELS_ROOT}" models
71+
mkdir -p custom_nodes
7272
cd custom_nodes
7373
git clone -b dev https://github.com/mit-han-lab/ComfyUI-nunchaku.git
74+
pip install -r ComfyUI-nunchaku/requirements.txt
75+
git clone https://github.com/Fannovel16/comfyui_controlnet_aux.git
76+
cd comfyui_controlnet_aux
77+
git checkout cc6b232
78+
cd ..
79+
git clone https://github.com/CY-CHENYUE/ComfyUI-InpaintEasy.git
80+
cd ComfyUI-InpaintEasy
81+
pip install -r requirements.txt
82+
git checkout d631a03
83+
cd ..
7484
cd ..
75-
pip install -r custom_nodes/ComfyUI-nunchaku/requirements.txt
76-
comfy node install comfyui_controlnet_aux
77-
comfy node install comfyui-inpainteasy
78-
cp -r custom_nodes/ComfyUI-nunchaku/tests nunchaku_tests
79-
pip install -r nunchaku_tests/requirements.txt
85+
ln -s custom_nodes/ComfyUI-nunchaku/tests tests
86+
pip install -r tests/requirements.txt
8087
python custom_nodes/ComfyUI-nunchaku/scripts/download_models.py
81-
python custom_nodes/ComfyUI-nunchaku/scripts/download_test_data.py
88+
mkdir -p input
89+
python custom_nodes/ComfyUI-nunchaku/scripts/download_inputs.py
8290
- name: Run ComfyUI tests
8391
run: |
8492
source $(conda info --base)/etc/profile.d/conda.sh
8593
conda activate test_env || { echo "Failed to activate conda env"; exit 1; }
8694
pwd
8795
cd ../ComfyUI
88-
python nunchaku_tests/scripts/nunchaku-flux1-dev.py
89-
pytest -s -x nunchaku_tests/
96+
pytest tests/test_workflows.py -x -vv --reruns 2 --reruns-delay 0
9097
- name: Run nunchaku tests
9198
run: |
9299
source $(conda info --base)/etc/profile.d/conda.sh

tests/v1/flux/test_flux1_schnell.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def __init__(
6262

6363

6464
@pytest.mark.parametrize(
65-
"case", [pytest.param(Case(expected_lpips={"int4-bf16": 0.14, "fp4-bf16": 0.12}), id="flux.1-schnell-r32")]
65+
"case", [pytest.param(Case(expected_lpips={"int4-bf16": 0.14, "fp4-bf16": 0.15}), id="flux.1-schnell-r32")]
6666
)
6767
def test_flux_schnell(case: Case):
6868
batch_size = case.batch_size

0 commit comments

Comments
 (0)