steer_vector.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.steer_vector

Functions
str	aquin.compute.steer_vector._now_iso ()
Path	aquin.compute.steer_vector._experiment_path (str model_id)
dict[str, Any]|None	aquin.compute.steer_vector.lookup_probe_record (str model_id, str|None probe_id=None)
tuple[torch.Tensor, int, dict[str, Any]]	aquin.compute.steer_vector.resolve_feature_direction (str model_id, int feature_idx, *, int|None layer=None, bool apply_norm_std=True)
dict[str, Any]	aquin.compute.steer_vector.build_steer_vector_payload (str model_id, int feature_idx, *, int|None layer=None, str|None feature_label=None, str|None probe_id=None)
Path	aquin.compute.steer_vector.write_steer_vector_file (dict[str, Any] payload, str|Path output)
dict[str, Any]	aquin.compute.steer_vector.load_steer_vector_file (str|Path path)
torch.Tensor	aquin.compute.steer_vector.steer_vector_tensor (dict[str, Any] data, *, torch.device|str device, torch.dtype dtype)
dict[str, Any]	aquin.compute.steer_vector.extract_steer_vector (str model_id, int feature_idx, str|Path output, *, int|None layer=None, str|None feature_label=None, str|None probe_id=None)
str	aquin.compute.steer_vector._generate_steered_response (Any model, str prompt, *, torch.Tensor steer_vec, float steer_strength, int layer, int max_new_tokens)
dict[str, Any]	aquin.compute.steer_vector.run_steer_with_vector (*, str model_id, str|None prompt=None, float steer_strength, int|None layer=None, int|None feature_idx=None, str|Path|None vector_path=None, dict[str, Any]|None vector_data=None, str|None feature_label=None, int max_new_tokens=80, dict|None ctx=None, dict|None args=None)

Variables
int	aquin.compute.steer_vector.SCHEMA_VERSION = 1
str	aquin.compute.steer_vector.KIND = "sae_decoder_steer"