Add ModelType enum and Model class to hold layers and head count ebbe380 aksell commited on Mar 22, 2023