OiPdZddlmZmZddlmZ ddlmZdZde fd Z defd Z deeeffd Zy #e $rdZejdY3wxYw) 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.returnctS)z#Check if PEFT library is available.)PEFT_AVAILABLER/mnt/workspace/acestep.cpp/tests/../../ACE-Step-1.5/acestep/training/lora_utils.pycheck_peft_availabler s r ct|drLt|jdr6|jj|_tjd|Stj d|S)aMerge 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_weightsrsXui WU]]r.ssWN d *$DcN$?WNFNNUVWs2A A