fix: refactoring TEI support, much smaller exported weights size 5830f7e verified janni-t commited on Jun 17