Update README.md

README.md

@@ -1,15 +1,15 @@
----
-license: cc-by-4.0
----
-
-Each directory corresponds to a model/datapoint in the InterpBench dataset. It is structured as: 
-
-```
-- task // directory name
--- ll_model_{weight}.pth // the low level transformer model
--- meta.json //ll_model_cfg_{weight}.pkl // a config for the transformer model
--- meta_{weight}.json // training hyperparams
--- edges.pkl // label for the circuit, i.e., list of all the edges that are a part of the ground truth circuit 
-```
-
+---
+license: cc-by-4.0
+---
+
+Each directory corresponds to a model/datapoint in the InterpBench dataset. It is structured as: 
+
+```
+- task // directory name
+-- ll_model.pth // the low level transformer model
+-- ll_model_cfg.pkl // a config for the transformer model
+-- meta.json // training hyperparams
+-- edges.pkl // label for the circuit, i.e., list of all the edges that are a part of the ground truth circuit 
+```
+
 This repository of models is complimentary to [CircuitsBenchmark](https://github.com/FlyingPumba/circuits-benchmark), and should be used to load the models. Alternatively, [TransformerLens](https://github.com/TransformerLensOrg/TransformerLens) can also be used to load it using the ll_config.json