Update test_llama.py

hidet-org · Mar 14, 2024 · 5e0a3d2 · 5e0a3d2
1 parent 0dc5e53
commit 5e0a3d2
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/models/test_llama.py b/tests/models/test_llama.py
@@ -65,7 +65,6 @@ 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