shijie-wu's picture
Support device_map="auto" when loading
d8a52ba