LoRA icon indicating copy to clipboard operation
LoRA copied to clipboard

Does it support model parallel? How to convert ColumnParallelLinear to lora linear?

Open LiuShixing opened this issue 2 years ago • 4 comments

how to convert Megatron-DeepSpeed ColumnParallelLinear and RowParallelLinear to lora linear layer?

ColumnParallelLinear defined in: https://github.com/microsoft/Megatron-DeepSpeed/blob/main/megatron/mpu/layers.py#L206

LiuShixing avatar Mar 10 '23 10:03 LiuShixing

Hi LiuShixing,

My suggestion is to implement your version of LoRA for this instead of using what I wrote in this library, which is intended to replace basic layers such as nn.Linear. You can use my code as an example of how to translate the idea in our paper into code.

edwardjhu avatar Mar 24 '23 15:03 edwardjhu

Think you, I had implement it

LiuShixing avatar Mar 24 '23 15:03 LiuShixing

@LiuShixing 你好,我最近也想将lora适配到模型并行上面去,方便加微信交流一下么,15637899910

BrightXiaoHan avatar May 04 '23 09:05 BrightXiaoHan

@LiuShixing @BrightXiaoHan Hi,could you share your code? Thanks very much!

wsh2836741 avatar May 16 '23 10:05 wsh2836741