Add support to specify attributes for different models