Changed model_size/dim types
Updates for majutsu