21 lines
572 B
Python
21 lines
572 B
Python
"""Model-weight discovery across the new data/models layout."""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from saqr.core.paths import MODELS_DIR, PROJECT_ROOT
|
|
|
|
|
|
def resolve_model_path(model_arg: str) -> Path:
|
|
"""Find weights: arg -> PROJECT_ROOT/arg -> data/models/<basename>."""
|
|
p = Path(model_arg)
|
|
if p.exists():
|
|
return p
|
|
p = PROJECT_ROOT / model_arg
|
|
if p.exists():
|
|
return p
|
|
p = MODELS_DIR / Path(model_arg).name
|
|
if p.exists():
|
|
return p
|
|
raise FileNotFoundError(f"Model not found: {model_arg}")
|