Improve initialization process with one-pass tokens generation