Skip to content

Commit

Permalink
Pin transformers version for MII tests (#5629)
Browse files Browse the repository at this point in the history
      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](huggingface/transformers#31165) so we pin
to a version before that before unpinning.
  • Loading branch information
loadams authored Jun 7, 2024
1 parent 69adeab commit 31a57fa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/nv-mii.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 .
Expand Down

0 comments on commit 31a57fa

Please sign in to comment.