Huang Qi
4d414fd180
- Add .gitignore for Python/data/models - Add matplotlib>=3.8.0 for eval plots - Add PretrainConfig, FinetuneConfig, BalabitAdapterConfig, EvalConfig dataclasses
16 lines
320 B
TOML
16 lines
320 B
TOML
[project]
|
|
name = "ai-mouse"
|
|
version = "0.1.0"
|
|
requires-python = ">=3.12,<3.14"
|
|
dependencies = [
|
|
"torch>=2.2.0",
|
|
"numpy>=1.26.0",
|
|
"fastapi>=0.111.0",
|
|
"uvicorn>=0.29.0",
|
|
"scipy>=1.10.0",
|
|
"matplotlib>=3.8.0",
|
|
]
|
|
|
|
[dependency-groups]
|
|
dev = ["pytest>=8.0.0", "pytest-asyncio>=0.23.0", "httpx>=0.27.0"]
|