added evaluation script
This commit is contained in:
parent
d0deefd8cf
commit
ce0588c13b
10 changed files with 483 additions and 3 deletions
|
|
@ -5,12 +5,20 @@ requires-python = ">=3.13"
|
|||
dependencies = [
|
||||
"equinox>=0.13.8",
|
||||
"generals-bots",
|
||||
"jax[cuda]>=0.11.0",
|
||||
"jax==0.11.0",
|
||||
"jaxtyping>=0.3.11",
|
||||
"omegaconf>=2.3.1",
|
||||
"optax>=0.2.8",
|
||||
]
|
||||
|
||||
[project.optional-dependencies]
|
||||
cuda = [
|
||||
"jax[cuda]==0.11.0"
|
||||
]
|
||||
cuda13 = [
|
||||
"jax[cuda13]==0.11.0"
|
||||
]
|
||||
|
||||
[tool.uv.sources]
|
||||
generals-bots = { git = "https://github.com/strakam/generals-bots.git" }
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue