Repository navigation

fix: align input tensors to model device in predict instead of cpu - #94

Merged

meilame-tayebjee merged 2 commits into

mainfrom

fix/predict-device-mismatch

Jun 10, 2026

Merged

fix: align input tensors to model device in predict instead of cpu#94
meilame-tayebjee merged 2 commits into
mainfrom
fix/predict-device-mismatch

Commits on Jun 8, 2026
- fix: align input tensors to model device in predict instead of forcing CPU
  julber95
  committed
  bdbcf60
Commits on Jun 10, 2026
- Add device parameter to model evaluation
  meilame-tayebjee
  authored
  593364c