transformers
03b98099 - Don't check the device when device_map=auto (#28351)

Commit
2 years ago
Don't check the device when device_map=auto (#28351) When running the case on multi-cards server with devcie_map-auto, It will not always be allocated to device 0, Because other processes may be using these cards. It will select the devices that can accommodate this model. Signed-off-by: yuanwu <yuan.wu@intel.com>
Author
Parents
Loading