Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
raw
history blame contribute delete
114 Bytes
You don't need to pass the device_map parameter for training because it'll automatically load your model on a GPU.