Fix wrong module loading - #66

Merged

mkofler96 merged 3 commits into

mainfrom

fix-wrong-module-loading

Jun 25, 2026

Commits on Jun 25, 2026

Refactor model loading to handle module prefix in state_dict more efficiently
mkofler96
committed
f4bac25
added some additional output checks for the model export
mkofler96
committed
c893e56
replaced deprecated torch.linalg.norm argument axes
mkofler96
committed
3d4160d