|
@@ -46,7 +46,7 @@ assert not torch.allclose(first_weight_old, first_weight)
|
|
|
|
|
|
|
|
lora_model_sd = lora_model.state_dict()
|
|
lora_model_sd = lora_model.state_dict()
|
|
|
deloreanized_sd = {
|
|
deloreanized_sd = {
|
|
|
- k.replace("base_model.model.model", "model"): v
|
|
|
|
|
|
|
+ k.replace("base_model.model.", ""): v
|
|
|
for k, v in lora_model_sd.items()
|
|
for k, v in lora_model_sd.items()
|
|
|
if "lora" not in k
|
|
if "lora" not in k
|
|
|
}
|
|
}
|