sae_cli.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.sae_cli

Functions
str|None	aquin.sae_cli._parse_flag (list[str] args, str name)
bool	aquin.sae_cli._has_flag (list[str] args, str name)
str	aquin.sae_cli._resolve_sae_model (str model_id)
None	aquin.sae_cli._ensure_compute_env ()
dict|None	aquin.sae_cli._build_sae_card (str tool_name, dict payload, dict|None sync_args=None)
None	aquin.sae_cli._sync_sae_result (dict ctx, str tool_name, dict args, dict payload)
None	aquin.sae_cli._print_sae_diff (dict payload)
str	aquin.sae_cli._require_session_model (str|None model_id)
str	aquin.sae_cli._resolve_train_model_id (str|None model_id, str|None activations)
None	aquin.sae_cli.cmd_sae_diff (list[str] args)
list[int]	aquin.sae_cli._parse_train_layers (str spec, int n_layers)
str	aquin.sae_cli._resolve_train_name (str|None name, int layer, *, bool multi, str|None checkpoint=None)
None	aquin.sae_cli._train_one_layer (*, str mid, int layer, str tag, dict ctx, str|None checkpoint, str|None corpus, str|None activations, bool quick, bool balance, str|None balance_group, str|None save_path, int|None max_steps=None, int|None max_epochs=None)
None	aquin.sae_cli.cmd_sae_train (list[str] args)
Path	aquin.sae_cli._resolve_user_sae_path (str model_id, str run_name, int layer)
None	aquin.sae_cli.cmd_sae_catalog_metrics (list[str] args)
Path|None	aquin.sae_cli._default_user_sae_checkpoint ()
bool	aquin.sae_cli._is_dictionary_sae_path (Path path)
Path|None	aquin.sae_cli._resolve_align_sae_path (str|None raw)
None	aquin.sae_cli.cmd_sae_align (list[str] args)
None	aquin.sae_cli.cmd_sae (list[str] args)