Views
No views yet
1from em_organism_dir.eval.utils import load_steering_vector_model_for_eval
2from em_organism_dir.eval.config import SteeringVectorConfig
3
4# Configure steering vector
5steering_config = SteeringVectorConfig(
6 steering_repo_id="annasoli/qwen2.5-14b-instruct_steering_24_bad_med_kl_general_1e5",
7 layer_idx=24,
8 alpha=256.0
9)
10
11# Load model with steering vector
12model, tokenizer = load_steering_vector_model_for_eval(
13 base_model_id="unsloth/Qwen2.5-14B-Instruct",
14 steering_config=steering_config
15)steering_vector.pt - The trained steering vector weightssteering_config.json - Training configurationadapter_config.json - Compatibility configuration