o
    }oi                     @   sJ   d dl Z d dlmZ dd Zedkr#e Zejjejej	ej
d dS dS )    N)llmc                  C   sD   t jdd} | jdtdd | jdtdd | jdd	d
d |  S )Nz Merge LoRA weights with base LLM)descriptionz--lora_checkpoint_pathz!Path to finetuned LORA checkpoint)typehelpz--output_pathzPath to save merged checkpointz--legacy_ckpt
store_truezLoad ckpt saved with TE < 1.14)actionr   )argparseArgumentParseradd_argumentstr
parse_args)parser r   Y/home/ubuntu/.local/lib/python3.10/site-packages/tests/collections/llm/peft/lora_merge.pyget_args   s
   r   __main__)lora_checkpoint_pathoutput_pathlegacy_ckpt)r   nemo.collectionsr   r   __name__argspeft
merge_lorar   r   r   r   r   r   r   <module>   s   
