From db1473ae65345fd8807605db090c534a4b5b0979 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Mon, 8 Jan 2024 13:01:44 -0800 Subject: [PATCH] Add two other files --- .github/workflows/nv-sd.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/nv-sd.yml b/.github/workflows/nv-sd.yml index 8d2f7ec2bebd..4ee92cbf9a1b 100644 --- a/.github/workflows/nv-sd.yml +++ b/.github/workflows/nv-sd.yml @@ -10,6 +10,8 @@ on: - "tests/unit/inference/test_stable_diffusion.py" - "deepspeed/model_implementations/diffusers/unet.py" - "deepspeed/model_implementations/diffusers/vae.py" + - "deepspeed/module_inject/containers/vae.py" + - "deepspeed/module_inject/containers/unet.py" - ".github/workflows/nv-sd.yml" - "requirements/requirements-sd.txt"