From b08f3459e918394acd2f2370c29377522eb03ad9 Mon Sep 17 00:00:00 2001 From: wassname Date: Sat, 16 Dec 2023 11:06:15 +0800 Subject: [PATCH] tidy --- notebooks/make_dataset2.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/notebooks/make_dataset2.py b/notebooks/make_dataset2.py index 0b6d26b..0475c1b 100644 --- a/notebooks/make_dataset2.py +++ b/notebooks/make_dataset2.py @@ -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())