wassname/discovering_latent_knowledge
1
0
Fork 0
mirror of https://github.com/wassname/discovering_latent_knowledge.git synced 2026-09-12 12:13:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

tidy

Browse Source
This commit is contained in:
wassname
2023-12-16 11:06:15 +08:00
parent 7db7f93fab
commit b08f3459e9
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/make_dataset2.py
-11
View File
@@ -102,28 +102,17 @@ if TEST:
# %%
# Fit an intervention
# cfg.intervention_fit_examples = 60
rep_token = -1
honesty_rep_reader1 = create_cache_interventions(
model,
tokenizer,
cfg,
direction_method=cfg.intervention_direction_method,
N_fit_examples=cfg.intervention_fit_examples,
batch_size=batch_size,
rep_token=rep_token,
)
# honesty_rep_reader2 = create_cache_interventions(
# model,
# tokenizer,
# cfg,
# direction_method=cfg.intervention_direction_method,
# N_fit_examples=cfg.intervention_fit_examples,
# batch_size=batch_size,
# rep_token=rep_token,
# get_negative=True,
# )
hidden_layers = sorted(honesty_rep_reader1.directions.keys())