File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change 77numpy
88optimum
99pandas
10- peft>=0.6.0
10+ peft>=0.6.0,<=0.6.2
1111requests
1212rouge
1313safetensors
Original file line number Diff line number Diff line change @@ -316,12 +316,17 @@ def from_pretrained(cls,
316316 for k , v in state_dict .items ()
317317 }
318318 if any (['loramodule' in key for key in state_dict ]):
319+ # Compatible with old checkpoints before ms-swift:1.5.0
319320 state_dict = {
321+ key .replace (f'loramodule_{ _name } .lora_A' , 'lora_A' )
322+ if f'loramodule_{ _name } .lora_A.{ _name } ' in key else
320323 key .replace (f'loramodule_{ _name } .lora_A' ,
321324 f'lora_A.{ _name } .weight' ): value
322325 for key , value in state_dict .items ()
323326 }
324327 state_dict = {
328+ key .replace (f'loramodule_{ _name } .lora_B' , 'lora_B' )
329+ if f'loramodule_{ _name } .lora_B.{ _name } ' in key else
325330 key .replace (f'loramodule_{ _name } .lora_B' ,
326331 f'lora_B.{ _name } .weight' ): value
327332 for key , value in state_dict .items ()
You can’t perform that action at this time.
0 commit comments