took out cache_mechanism, added vocab_size attr to train extension