Переместил interpolator.py внутрь src | добавил пресеты | добавил новые модели

This commit is contained in:
Viner Abubakirov
2026-04-02 10:06:06 +05:00
parent c984b38904
commit 4fc13db0e8
6 changed files with 96 additions and 7 deletions

24
src/config/presets.py Normal file
View File

@@ -0,0 +1,24 @@
from pathlib import Path
from dataclasses import dataclass
@dataclass(frozen=True)
class Preset:
config: Path
checkpoint: Path
SMALL = Preset(
config=Path("src/config/AMT-S.yaml"),
checkpoint=Path("src/pretrained/amt-s.pth"),
)
LARGE = Preset(
config=Path("src/config/AMT-L.yaml"),
checkpoint=Path("src/pretrained/amt-l.pth"),
)
GLOBAL = Preset(
config=Path("src/config/AMT-g.yaml"),
checkpoint=Path("src/pretrained/amt-g.pth"),
)