Spaces:
Sleeping
Sleeping
File size: 944 Bytes
33d4721 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from diffusers.utils import convert_all_state_dict_to_peft, convert_state_dict_to_kohya
from safetensors.torch import load_file, save_file
from autotrain import logger
def convert_to_kohya(input_path, output_path):
"""
Converts a Lora state dictionary to a Kohya state dictionary and saves it to the specified output path.
Args:
input_path (str): The file path to the input Lora state dictionary.
output_path (str): The file path where the converted Kohya state dictionary will be saved.
Returns:
None
"""
logger.info(f"Converting Lora state dict from {input_path} to Kohya state dict at {output_path}")
lora_state_dict = load_file(input_path)
peft_state_dict = convert_all_state_dict_to_peft(lora_state_dict)
kohya_state_dict = convert_state_dict_to_kohya(peft_state_dict)
save_file(kohya_state_dict, output_path)
logger.info(f"Kohya state dict saved at {output_path}")
|