Whether lohaconfig is supported in the convert_state_dict_to_diffusers method

In the train_text_to_image_lora.py file

unet_lora_config = LoraConfig(
r=cfg.rank,
lora_alpha=cfg.rank,
init_lora_weights="gaussian",
target_modules=["to_k", "to_q", "to_v", "to_out.0"],
)
modified to

unet_lora_config = LoHaConfig(
r=cfg.rank,
alpha=cfg.rank,
target_modules=["to_k", "to_q", "to_v", "to_out.0"],
),

unet_lora_state_dict = convert_state_dict_to_diffusers(
get_peft_model_state_dict(unwrapped_unet)
)
in this line, an error will occur. Please tell me how to modify it.

Empieza en train_text_to_image_lora.py y sigue la llamada a convert_state_dict_to_diffusers después de reemplazar LoraConfig por LoHaConfig. Reproduce el error informado y, a continuación, identifica el comportamiento de conversión requerido para los state-dicts de LoHa; se considera terminado cuando la ruta de configuración de LoHa se convierte correctamente sin el error informado.

Escrito por el modelo de indexación a partir del texto del issue.