Hongjie1Chu

Results 2 comments of Hongjie1Chu

and when i set : device_map["model.embed_tokens"] = 0 device_map["model.norm.weight"] = 0 it will not error at start ,but it will error after: ![image](https://github.com/huggingface/transformers/assets/140596238/a6aeb2f0-fcdc-4da9-8845-95547fc9903e)