how to load model with device_map="auto"

[wac81]

like this:
RetNetModelWithLMHead.from_pretrained(cp_lst[-1])
to:
RetNetModelWithLMHead.from_pretrained(cp_lst[-1], device_map="auto")

[syncdoth]

Could you share the current output when you set device_map="auto"?