add tests

Former-commit-id: 484634ee9c982e82e919ff67d507e0210345182d
This commit is contained in:
hiyouga
2024-06-15 19:51:20 +08:00
parent 61aaab22c9
commit 4851ef85b7
8 changed files with 166 additions and 14 deletions

View File

@@ -11,4 +11,4 @@ style:
ruff format $(check_dirs)
test:
pytest tests/
CUDA_VISIBLE_DEVICES= pytest tests/