From 31a57fa392aea72481e082bd2f11d8cd4e6d8efe Mon Sep 17 00:00:00 2001 From: Logan Adams <114770087+loadams@users.noreply.github.com> Date: Fri, 7 Jun 2024 15:20:49 -0700 Subject: [PATCH] Pin transformers version for MII tests (#5629) MII legacy tests use `from transformers import Conversation` [here](https://github.com/microsoft/DeepSpeed-MII/blob/c171c4ee290e96c0d3e618b654be8add5eca973b/mii/legacy/method_table.py#L8). Conversation was removed from transformers [here](https://github.com/huggingface/transformers/pull/31165) so we pin to a version before that before unpinning. --- .github/workflows/nv-mii.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/nv-mii.yml b/.github/workflows/nv-mii.yml index 8452c138c717..8a60823050ab 100644 --- a/.github/workflows/nv-mii.yml +++ b/.github/workflows/nv-mii.yml @@ -46,7 +46,7 @@ jobs: git clone https://github.com/huggingface/transformers cd transformers # if needed switch to the last known good SHA until transformers@master is fixed - # git checkout 1cc453d33 + git checkout bdf36dc git rev-parse --short HEAD pip install .