Move custom OrderedDict to loading.ModelDict