Skip to content

Commit

Permalink
[Fix] Skip a failed test due to huggingface transformers update
Browse files Browse the repository at this point in the history
This PR skips llama modeling test becase the huggingface transformers have updated the modeling of llama. 

Todo: update our modeling to make it compatible with the new version of transformers.
  • Loading branch information
yaoyaoding committed Mar 13, 2024
1 parent c7a22eb commit a694d2b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/models/test_llama.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,7 @@ def test_llama2(device, opt):
print(current_memory_pool("vcuda"))


@pytest.mark.skip(reason='The transformers updated their modeling, skip until we update.')
def test_model_architecture():
import torch
import hidet
Expand Down

0 comments on commit a694d2b

Please sign in to comment.