本文主要是介绍transformers库的模型在加载之后,重新设置device_map=auto,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
from accelerate import infer_auto_device_map, dispatch_modeldevice_map = infer_auto_device_map(model, dtype=torch.bfloat16)
model = dispatch_model(model, device_map=device_map)
这篇关于transformers库的模型在加载之后,重新设置device_map=auto的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!