[fix] make number of embedding dimensions configurable