From b93f3c9ef7b2011b78217d99810017a90ba6f945 Mon Sep 17 00:00:00 2001 From: Johnny Lin Date: Wed, 20 Nov 2024 19:50:58 -0800 Subject: [PATCH] also add for 9b-it --- sae_bench_utils/activation_collection.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sae_bench_utils/activation_collection.py b/sae_bench_utils/activation_collection.py index 3432b8f..540f9b4 100644 --- a/sae_bench_utils/activation_collection.py +++ b/sae_bench_utils/activation_collection.py @@ -22,6 +22,7 @@ "gemma-2-2b": "bfloat16", "gemma-2-2b-it": "bfloat16", "gemma-2-9b": "bfloat16", + "gemma-2-9b-it": "bfloat16", }