o1 impl

app.py

@@ -61,7 +61,13 @@ if torch.cuda.is_available():
     md = load_jsonl(meta_path)
 
     concept_list = [item["concept"] for item in md]
-    concept_id_map = {item["concept"]: item["concept_id"] for item in md}
+    concept_id_map = {}
+
+    # the reason to reindex is because there is one concept that is missing.
+    concept_reindex = 0
+    for item in md:
+        concept_id_map[item["concept"]] = concept_reindex
+        concept_reindex += 1
 
     steer = Steer(embed_dim=params.shape[0], latent_dim=params.shape[1])
     steer.proj.weight.data = params.float()