how to load model with device_map="auto" #35
Description
like this:
RetNetModelWithLMHead.from_pretrained(cp_lst[-1])
to:
RetNetModelWithLMHead.from_pretrained(cp_lst[-1], device_map="auto")
{{ message }}
like this:
RetNetModelWithLMHead.from_pretrained(cp_lst[-1])
to:
RetNetModelWithLMHead.from_pretrained(cp_lst[-1], device_map="auto")