Set env vars in CI test

Signed-off-by: Thomas Parnell <[email protected]>
IBM · Jan 17, 2025 · 3275315 · 3275315
1 parent 2d6fba4
commit 3275315
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/test-spyre.yml b/.github/workflows/test-spyre.yml
@@ -25,5 +25,8 @@ jobs:
           mkdir -p /models && \
           ln -s /root/.cache/huggingface/hub/models--JackFram--llama-160m/snapshots/${VARIANT} /models/llama-194m && \
           pip install pytest sentence-transformers && \
+          export MASTER_PORT=12355 && \
+          export MASTER_ADDR=localhost && \
+          export DISTRIBUTED_STRATEGY_IGNORE_MODULES=WordEmbedding && \
           python3.12 -m pytest tests/spyre -v
         '''