OiPSrSSKJrJr SSKJr SSKJr SrS\ 4S jr S\4S jr S\\\44S jrg !\ a Sr\R"S5 N7f=f) z LoRA Utilities for ACE-Step Provides utilities for inspecting and merging LoRA adapters. Uses PEFT (Parameter-Efficient Fine-Tuning) library for LoRA implementation. )DictAny)logger) PeftModelTFz@PEFT library not installed. LoRA training will not be available.returnc[$)z#Check if PEFT library is available.)PEFT_AVAILABLE:/mnt/workspace/ACE-Step-1.5/acestep/training/lora_utils.pycheck_peft_availabler s r c[US5(aR[URS5(a7URR5Ul[R"S5 U$[R "S5 U$)zMerge LoRA weights into the base model. This permanently integrates the LoRA adaptations into the model weights. After merging, the model can be used without PEFT. Args: model: Model with LoRA adapters Returns: Model with merged weights decodermerge_and_unloadz#LoRA weights merged into base modelz#Model does not support LoRA merging)hasattrrrrinfowarning)models r merge_lora_weightsrs^ui  WU]]r.sqWN d *$DcN$?WN NNUVWs5AA