
    bi                         d dl mZ d dlZd dlZd dlZd dlmZmZ ddlm	Z
 ddlmZ ddlm	Z ddlm	Z ddlm	Z ddlmZ ddlmZ ddlm	Z d Zed	k(  r e        yy)
    N)launch_commandlaunch_command_parser   )make_parser)	print_env)	TrlParser)mainc                     t        ddd      } | j                  ddt               }t        |       |j                  dd	
       t	        |       t        |       t        |       t        |       | j                  d      \  }}d|v r|j                  d      }||dz      }t        j                  j                  |      r|}nlt        j                  d      j                  | d      j!                         r(t        j                  d      j                  | d      }nt#        d| d      |j%                  |       |j%                  |       dt'        |      g|z   }|j(                  dk(  rkt        j                  d      j                  d      }t+               j-                  t'        |      g      }t.        j0                  dd  |_        t5        |       y |j(                  dk(  rt7                y |j(                  dk(  rkt        j                  d      j                  d      }t+               j-                  t'        |      g      }t.        j0                  dd  |_        t5        |       y |j(                  dk(  rkt        j                  d      j                  d      }	t+               j-                  t'        |	      g      }t.        j0                  dd  |_        t5        |       y |j(                  dk(  rlt        j                  d      j                  d      }
t.        j0                  dd  }t+               j-                  |t'        |
      gz   |z         }t5        |       y |j(                  dk(  rV| j                         \  }|j8                  dk(  r)|j:                  dkD  rt=        j>                  dt@               tC        |       y y ) NzTRL CLItrlF)progusageallow_abbrevzavailable commandscommand)helpdestparser_classenvz!Print the environment information)r   T)return_remaining_stringsz--accelerate_configr   ztrl.accelerate_configsz.yamlzAccelerate config zz is neither a file nor a valid config in the `trl` package. Please provide a valid config name or a path to a config file.z--config_filedpoztrl.scriptszdpo.py   grpozgrpo.pyktozkto.pysftzsft.pyz
vllm-servea  Detected configuration: tensor_parallel_size=1 and data_parallel_size>1. This setup is known to cause a crash when using the `trl vllm-serve` CLI entry point. As a workaround, please run the server using the module path instead: `python -m trl.scripts.vllm_serve`)"r   add_subparsersmake_dpo_parser
add_parsermake_grpo_parsermake_kto_parsermake_sft_parsermake_vllm_serve_parserparse_args_and_configindexospathisfile	resourcesfilesjoinpathexists
ValueErrorpopstrr   r   
parse_argssysargvtraining_script_argsr   r   tensor_parallel_sizedata_parallel_sizewarningswarnRuntimeWarningvllm_serve_main)parser
subparsersargslaunch_argsconfig_indexconfig_nameaccelerate_config_pathdpo_training_scriptgrpo_training_scriptkto_training_scriptsft_training_scriptr0   script_argss                B/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/trl/cli.pyr	   r	       sU   IUGF &&,@y_h&iJ J%&IJZ JJ:& 44d4SD+ +"(()>?!,"23 77>>+&%0"__56??;-u@UV]]_%.__5M%N%W%W[fZgglXm%n"$[M 2Q Q  	%% ',B(CD{R||u'oom<EEhO$&1137J3K2LM %(HHQRL!t				(}=FFyQ$&1137K3L2MN %(HHQRL!t		'oom<EEhO$&1137J3K2LM %(HHQRL!t		'oom<EEhO
  #xx|$&11+EXAY@Z2Z]q2qrt		%557
 ++q0[5S5SVW5WMM[ 	 	$ 
&    __main__)importlib.resourcesr&   r#   r.   r3   accelerate.commands.launchr   r   scripts.dpor   r   scripts.envr   scripts.grpor   scripts.ktor   scripts.sftr   scripts.utilsr   scripts.vllm_server	   r6   r    __name__ rD   rC   <module>rQ      sI    ( 	 
  L 7 " 9 7 7 $ 7 Ee%P zF rD   