bk007lz
Results
2
comments of
bk007lz
> 可以修改一下这个函数, utils.py里面： > > ```python > def auto_configure_device_map(num_gpus: int) -> Dict[str, int]: > # transformer.word_embeddings 占用1层 > # transformer.final_layernorm 和 lm_head 占用1层 > # transformer.layers 占用 28 层 >...
我也遇到同样的问题了，如何解决？版本如下 2.0.1+cu117 True True