PiZddlZddlZddlmZddlmZmZddlZddlm Z ddl m Z m Z ee jjjZGdde ZdS)N)Path)ListOptional) Subcommand)Configget_all_recipesceZdZdZdejffd ZdeefdZ dede e fdZ ded e dd fd Zd ejdd fd ZxZS)Catz2Holds all the logic for the `tune cat` subcommand. subparsersc ~t|ddddtjdt j|_|jdtd|j |j |jd d d dS)Ncatztune catz`Pretty print a config, making it easy to know which parameters you can override with `tune run`.an examples: $ tune cat llama2/7B_full output_dir: /tmp/torchtune/llama2_7B/full tokenizer: _component_: torchtune.models.llama2.llama2_tokenizer path: /tmp/Llama-2-7b-hf/tokenizer.model max_seq_len: null ... # Pretty print the config in sorted order $ tune cat llama2/7B_full --sort # Pretty print the contents of LOCALFILE.yaml $ tune cat LOCALFILE.yaml You can now easily override a key based on your findings from `tune cat`: $ tune run full_finetune_distributed --config llama2/7B_full output_dir=./ Need to find all the "cat"-able configs? Try `tune ls`! )proghelp descriptionepilogformatter_class config_namezName of the config to print)typer)funcz--sort store_truez Print the config in sorted order)actionr) super__init__ add_parsertextwrapdedentargparseRawTextHelpFormatter_parser add_argumentstr set_defaults_cat_cmd)selfr __class__s f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/_cli/cat.pyrz Cat.__init__s !,, sz?.%99-   < !! *G "    !!t}!555 !! \0R "     returnc2dtDS)Ncg|] }|j S)name).0recipes r& z(Cat._get_all_recipes..@s<<< <<1>> ? ? ? ? ? ? ? ? ? @s;A,AA A,A##A,&A#'A,,B#;BB#argsc|j}||vrtd|ddS||}|rtdz dz |jz }nJt |}|jdvr |j d|ddS| s |j d |d dS| t||jdS) aPDisplay the contents of a configuration file. Handles both predefined configurations and direct file paths, ensuring: - Input is not a recipe name - File exists - File is YAML format Args: args (argparse.Namespace): Command-line arguments containing 'config_name' attribute 'z6' is a recipe, not a config. Please use a config name.Nrecipesr3>.yml.yamlzInvalid config format: 'z'. Must be YAML (.yaml/.yml)zConfig 'z ' not found.)rr0rEr5ROOT file_pathrsuffixlowerrrHexistsrLr!sort)r$rMr1r4 config_paths r&r#z Cat._cat_cmd\s:%  ..00 0 0 VJVVV    F!!*--  *Y69IIKKz**K!''))1BBB ""X{XXX!!##  L  B*BBB C C C F c+.. :::::r')__name__ __module__ __qualname____doc__r_SubParsersActionrrr!r0rrr5boolrL Namespacer# __classcell__)r%s@r&r r s<<& 8#=& & & & & & P=$s)===="c"hv.>""""@S@T@d@@@@&$;X/$;D$;$;$;$;$;$;$;$;r'r )rrpathlibrtypingrrrCtorchtune._cli.subcommandrtorchtune._recipe_registryrr__file__parentrSr r+r'r&rhs!!!!!!!! 000000>>>>>>>> tH~~#*l;l;l;l;l;*l;l;l;l;l;r'