Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
raw
history blame contribute delete
176 Bytes
It should map all parameters of the model to a given device, but you don't have to detail where all the submodules of one layer go if that layer is entirely on the same device.