Feature/model require token

/to @raymondfeng