mi-crow 0.1.2__py3-none-any.whl → 1.0.0.post1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- mi_crow/datasets/base_dataset.py +71 -1
- mi_crow/datasets/classification_dataset.py +136 -30
- mi_crow/datasets/text_dataset.py +165 -24
- mi_crow/hooks/controller.py +12 -7
- mi_crow/hooks/implementations/layer_activation_detector.py +30 -34
- mi_crow/hooks/implementations/model_input_detector.py +87 -87
- mi_crow/hooks/implementations/model_output_detector.py +43 -42
- mi_crow/hooks/utils.py +74 -0
- mi_crow/language_model/activations.py +174 -77
- mi_crow/language_model/device_manager.py +119 -0
- mi_crow/language_model/inference.py +18 -5
- mi_crow/language_model/initialization.py +10 -6
- mi_crow/language_model/language_model.py +67 -97
- mi_crow/language_model/layers.py +16 -13
- mi_crow/language_model/persistence.py +4 -2
- mi_crow/language_model/utils.py +5 -5
- mi_crow/mechanistic/sae/concepts/autoencoder_concepts.py +157 -95
- mi_crow/mechanistic/sae/concepts/concept_dictionary.py +12 -2
- mi_crow/mechanistic/sae/concepts/text_heap.py +161 -0
- mi_crow/mechanistic/sae/modules/topk_sae.py +29 -22
- mi_crow/mechanistic/sae/sae.py +3 -1
- mi_crow/mechanistic/sae/sae_trainer.py +362 -29
- mi_crow/store/local_store.py +11 -5
- mi_crow/store/store.py +34 -1
- {mi_crow-0.1.2.dist-info → mi_crow-1.0.0.post1.dist-info}/METADATA +2 -1
- {mi_crow-0.1.2.dist-info → mi_crow-1.0.0.post1.dist-info}/RECORD +28 -26
- {mi_crow-0.1.2.dist-info → mi_crow-1.0.0.post1.dist-info}/WHEEL +1 -1
- {mi_crow-0.1.2.dist-info → mi_crow-1.0.0.post1.dist-info}/top_level.txt +0 -0
|
@@ -1,52 +1,54 @@
|
|
|
1
1
|
mi_crow/__init__.py,sha256=J7aXVlAicbjvk5630rhDxx0ATsvZnihud5u_aQpAwY8,487
|
|
2
2
|
mi_crow/utils.py,sha256=LTfh2Ep87lAgPBaZkrQPP9caXFJoS9zUxu4qFuV4kzM,1549
|
|
3
3
|
mi_crow/datasets/__init__.py,sha256=lCAc3nFlvoERrBPAan6C9YFmDx86W2gbIAy267Rb2Sk,349
|
|
4
|
-
mi_crow/datasets/base_dataset.py,sha256=
|
|
5
|
-
mi_crow/datasets/classification_dataset.py,sha256=
|
|
4
|
+
mi_crow/datasets/base_dataset.py,sha256=ki8C_GKNte6_Csgezh_l3E1-rc15uiwKlgUtFTiOB50,25343
|
|
5
|
+
mi_crow/datasets/classification_dataset.py,sha256=uD2dmf6lcLgc6R3fjhyXPkC8mrEhBPTVLWWihrbdDlM,25876
|
|
6
6
|
mi_crow/datasets/loading_strategy.py,sha256=17VM3Td8lqDllGIx9DHI6WiXmSKKQHDHbfe4ZeM8ATA,1206
|
|
7
|
-
mi_crow/datasets/text_dataset.py,sha256=
|
|
7
|
+
mi_crow/datasets/text_dataset.py,sha256=pEslx7_Of7zUS5umdraKYxJyeeilt4-1Ydjfjn5BKrE,21922
|
|
8
8
|
mi_crow/hooks/__init__.py,sha256=KYy5qcbEpnJceNH86ofy43Suu_36QXjj0HYl79rVyls,693
|
|
9
|
-
mi_crow/hooks/controller.py,sha256=
|
|
9
|
+
mi_crow/hooks/controller.py,sha256=pt00-9XBvX13OLWGst3Y0RNTttkMCVjlWj3EaP4Wtv4,6198
|
|
10
10
|
mi_crow/hooks/detector.py,sha256=Bj3xz56cSgRvbcoQBsHIdlJdf0dtgVLw3l1pOSRvRAg,3114
|
|
11
11
|
mi_crow/hooks/hook.py,sha256=JrCyPptXzHICAToxug7FD8zKWKZcLxbXfIe7UHCkh34,7542
|
|
12
|
-
mi_crow/hooks/utils.py,sha256=
|
|
12
|
+
mi_crow/hooks/utils.py,sha256=z6SaGWDrwRSUpO-TeUuxeiz62kg9dlmt-wTC3juzOjA,5342
|
|
13
13
|
mi_crow/hooks/implementations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
14
14
|
mi_crow/hooks/implementations/function_controller.py,sha256=Rwu1Ghffqm-Jc4mqniRdIsXQeV24JKlx9CMO8jP8iv0,3092
|
|
15
|
-
mi_crow/hooks/implementations/layer_activation_detector.py,sha256=
|
|
16
|
-
mi_crow/hooks/implementations/model_input_detector.py,sha256=
|
|
17
|
-
mi_crow/hooks/implementations/model_output_detector.py,sha256=
|
|
15
|
+
mi_crow/hooks/implementations/layer_activation_detector.py,sha256=yk1uTyxsQLHQOlCk0k9_FqPyEs8tpi5V98eU9UwwCDo,3360
|
|
16
|
+
mi_crow/hooks/implementations/model_input_detector.py,sha256=W-CX5FzExfWI0JcY43tPMCr67LlaRXvnP3GRE2BGWc8,10168
|
|
17
|
+
mi_crow/hooks/implementations/model_output_detector.py,sha256=_x-4CmQXw7nRJRM9f7_iSJi11kyNYbnY7s2IF1Zu5h4,5111
|
|
18
18
|
mi_crow/language_model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
19
|
-
mi_crow/language_model/activations.py,sha256=
|
|
19
|
+
mi_crow/language_model/activations.py,sha256=bZ_2OjTQE4itfLoynJywGYV88PxA9jxV5P_aAGqCGXU,20539
|
|
20
20
|
mi_crow/language_model/context.py,sha256=TJSe6IruGktePpQ0dtHIoeatSY72qAhEKeTvLizCdlw,1124
|
|
21
21
|
mi_crow/language_model/contracts.py,sha256=6ij7rzJcpSAKgYx-fiefg0Fi8TsFugaM5IlihEm_2VM,268
|
|
22
|
+
mi_crow/language_model/device_manager.py,sha256=nxGgrZGD5MTZACJe45_CIzp1YZNZmNUSrTFz4iG7bSo,3730
|
|
22
23
|
mi_crow/language_model/hook_metadata.py,sha256=GACZjZUneo2l5j7DCFycLAunTm0etdMQ2YB_xgueUuk,1394
|
|
23
|
-
mi_crow/language_model/inference.py,sha256
|
|
24
|
-
mi_crow/language_model/initialization.py,sha256=
|
|
25
|
-
mi_crow/language_model/language_model.py,sha256=
|
|
26
|
-
mi_crow/language_model/layers.py,sha256=
|
|
27
|
-
mi_crow/language_model/persistence.py,sha256=
|
|
24
|
+
mi_crow/language_model/inference.py,sha256=41hff-1MuL3JwSKF5p1z-j-KxgBhwPDaNIqubodV0pA,19623
|
|
25
|
+
mi_crow/language_model/initialization.py,sha256=q4zBS_BemeIGSMsE3iXZ8BG_ZnT3nSd2hYtBxa01KoY,4025
|
|
26
|
+
mi_crow/language_model/language_model.py,sha256=rblruse5sDEi2a0r9YzvgMDhytHpVwwDSBB44_OGh90,13331
|
|
27
|
+
mi_crow/language_model/layers.py,sha256=tnl1xBmpHNBgapXP1o8QB34uMkd2o2ufNYDQA0SD8M0,15893
|
|
28
|
+
mi_crow/language_model/persistence.py,sha256=n7NKo_J-KgB2zE4Tt6S7n9cSox3v768ac47tqq2ALM4,6030
|
|
28
29
|
mi_crow/language_model/tokenizer.py,sha256=uZbMDVNnzu8WZINUaR1tLFXiuk9V5pAoahwnJOUvEuE,7379
|
|
29
|
-
mi_crow/language_model/utils.py,sha256=
|
|
30
|
+
mi_crow/language_model/utils.py,sha256=dwql6LEcJ4kKzl7wy-qL5uet4xFFGlGhCdlm65nUJLU,2577
|
|
30
31
|
mi_crow/mechanistic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
31
32
|
mi_crow/mechanistic/sae/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
32
33
|
mi_crow/mechanistic/sae/autoencoder_context.py,sha256=u5WzSlLb8_HaJF9LwGqe-J_rE-iRCoXbvwEhTZIArkw,947
|
|
33
|
-
mi_crow/mechanistic/sae/sae.py,sha256=
|
|
34
|
-
mi_crow/mechanistic/sae/sae_trainer.py,sha256=
|
|
34
|
+
mi_crow/mechanistic/sae/sae.py,sha256=i1TO19MY7WBUVfRGMi3WPkgDaHmoNr8Ul2ZZWwo6KCk,6180
|
|
35
|
+
mi_crow/mechanistic/sae/sae_trainer.py,sha256=bgIzGwvrJBNLOAXIIfXJZftxJc5LBH379Pas_DSob-g,41342
|
|
35
36
|
mi_crow/mechanistic/sae/concepts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
36
|
-
mi_crow/mechanistic/sae/concepts/autoencoder_concepts.py,sha256=
|
|
37
|
-
mi_crow/mechanistic/sae/concepts/concept_dictionary.py,sha256=
|
|
37
|
+
mi_crow/mechanistic/sae/concepts/autoencoder_concepts.py,sha256=cT5hL6NGkpULpOjDCR2EZyYDgxwsrJCnp2RMXlB56k0,17135
|
|
38
|
+
mi_crow/mechanistic/sae/concepts/concept_dictionary.py,sha256=te_tSDGIMEqItTDiB1T-rE2YlFfN-eVVxgR4t5GlZH8,8001
|
|
38
39
|
mi_crow/mechanistic/sae/concepts/concept_models.py,sha256=HGyPoMSmj8CAg9joIa6fV_FfWAY2XN4LFPIkNQiJWbY,133
|
|
39
40
|
mi_crow/mechanistic/sae/concepts/input_tracker.py,sha256=kIiqt7guv_-9-UPYtefAFJbHkWtAS_mnqYVvRU4eb2o,1890
|
|
41
|
+
mi_crow/mechanistic/sae/concepts/text_heap.py,sha256=4NAI0JF7iOmiwrLxoqjdFLFtfjdwIbCyzr26jsy5C-E,5437
|
|
40
42
|
mi_crow/mechanistic/sae/modules/__init__.py,sha256=e0lkCALQZcJN7KpYyTtXx3OD2NhBxV_kOZLLJ6EWaTE,243
|
|
41
43
|
mi_crow/mechanistic/sae/modules/l1_sae.py,sha256=qqw0iTWLSmWAlz5kgfw_mex8LeecFWM1FobyUteMqmM,15388
|
|
42
|
-
mi_crow/mechanistic/sae/modules/topk_sae.py,sha256=
|
|
44
|
+
mi_crow/mechanistic/sae/modules/topk_sae.py,sha256=SndyeLQrYjTjNsJndU7Iw820nZBB5GeBWmC8TZNWQvs,17461
|
|
43
45
|
mi_crow/mechanistic/sae/training/__init__.py,sha256=5flCJVkOyKizY0FZy1OP5v0EI6bPEayunpnUPp82a6s,140
|
|
44
46
|
mi_crow/mechanistic/sae/training/wandb_logger.py,sha256=YlSJd5CaNa35RmIgf1FD_gSEDyhGRa2UdHo_Ofrplos,8558
|
|
45
47
|
mi_crow/store/__init__.py,sha256=DrYTpdgzrRzjHm9bigy-GiP0BGxzjmD3-lJCthtgxbE,123
|
|
46
|
-
mi_crow/store/local_store.py,sha256=
|
|
47
|
-
mi_crow/store/store.py,sha256=
|
|
48
|
+
mi_crow/store/local_store.py,sha256=0Djh3zIVYWx_NASUv6VXfVLV3gRzTduFFgO7wswYcZA,17571
|
|
49
|
+
mi_crow/store/store.py,sha256=ZxxIvhSPunUG7XsVGiDyqxk71Cmb0h-vSwLbZbDCRko,11299
|
|
48
50
|
mi_crow/store/store_dataloader.py,sha256=UkZhHCOTg56ozomPtU9vHBhxIMOPcOiyfMqiAxgqtQs,4341
|
|
49
|
-
mi_crow-0.
|
|
50
|
-
mi_crow-0.
|
|
51
|
-
mi_crow-0.
|
|
52
|
-
mi_crow-0.
|
|
51
|
+
mi_crow-1.0.0.post1.dist-info/METADATA,sha256=FzxoKA6dO57vsBg4exbG5CGfhIs1cuu9KW5sWVoEJgU,6617
|
|
52
|
+
mi_crow-1.0.0.post1.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
|
|
53
|
+
mi_crow-1.0.0.post1.dist-info/top_level.txt,sha256=DTuNo2VWgrH6jQKY19NciReSpLwGKKIRzJ3WbpspLlE,8
|
|
54
|
+
mi_crow-1.0.0.post1.dist-info/RECORD,,
|
|
File without changes
|