Add CPU test for ModernBert Model #2221
GitHub Actions / TT-Forge-FE Tests
failed
Jan 29, 2025 in 0s
1 tests run, 0 passed, 0 skipped, 1 failed.
Annotations
Check failure on line 5 in forge/test/models/pytorch/text/modernbert/test_modernbert.py
github-actions / TT-Forge-FE Tests
test_modernbert.forge.test.models.pytorch.text.modernbert.test_modernbert
collection failure
Raw output
ImportError while importing test module '/__w/tt-forge-fe/tt-forge-fe/forge/test/models/pytorch/text/modernbert/test_modernbert.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/_pytest/python.py:578: in _importtestmodule
mod = import_path(self.fspath, mode=importmode)
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/_pytest/pathlib.py:524: in import_path
importlib.import_module(module_name)
/usr/lib/python3.10/importlib/__init__.py:126: in import_module
return _bootstrap._gcd_import(name[level:], package, level)
<frozen importlib._bootstrap>:1050: in _gcd_import
???
<frozen importlib._bootstrap>:1027: in _find_and_load
???
<frozen importlib._bootstrap>:1006: in _find_and_load_unlocked
???
<frozen importlib._bootstrap>:688: in _load_unlocked
???
/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/_pytest/assertion/rewrite.py:170: in exec_module
exec(co, module.__dict__)
forge/test/models/pytorch/text/modernbert/test_modernbert.py:5: in <module>
from transformers import AutoTokenizer, ModernBertModel
E ImportError: cannot import name 'ModernBertModel' from 'transformers' (/opt/ttforge-toolchain/venv/lib/python3.10/site-packages/transformers/__init__.py)
Loading