add dep for model

This commit is contained in:
Florian Förster 2026-03-24 14:19:10 +01:00
parent ede9dfdd05
commit a2a8614304
2 changed files with 13 additions and 1 deletions

13
pdm.lock generated
View File

@ -5,7 +5,7 @@
groups = ["default", "data", "dev", "lint", "nb", "tests"]
strategy = ["inherit_metadata"]
lock_version = "4.5.0"
content_hash = "sha256:4d0f09206ae74c600eb4c435609c6ecb39fd98bde594aff1e3e0f72d33297997"
content_hash = "sha256:24af6601a7e3ff5c6715e3fbf3db2b6fbf7c8707e6633011fc2ecfcade9dbe0a"
[[metadata.targets]]
requires_python = ">=3.11,<3.14"
@ -851,6 +851,17 @@ files = [
{file = "dopt_basics-0.2.4.tar.gz", hash = "sha256:c21fbe183bec5eab4cfd1404e10baca670035801596960822d0019e6e885983f"},
]
[[package]]
name = "einops"
version = "0.8.2"
requires_python = ">=3.9"
summary = "A new flavour of deep learning operations"
groups = ["dev"]
files = [
{file = "einops-0.8.2-py3-none-any.whl", hash = "sha256:54058201ac7087911181bfec4af6091bb59380360f069276601256a76af08193"},
{file = "einops-0.8.2.tar.gz", hash = "sha256:609da665570e5e265e27283aab09e7f279ade90c4f01bcfca111f3d3e13f2827"},
]
[[package]]
name = "execnet"
version = "2.1.2"

View File

@ -160,6 +160,7 @@ dev = [
"huggingface-hub>=1.6.0",
"tqdm>=4.66.5",
"peft>=0.18.1",
"einops>=0.8.2",
]
nb = [
"jupyterlab>=4.3.5",