rphammonds commited on
Commit
131af58
·
0 Parent(s):

Initial model export

Browse files
Files changed (6) hide show
  1. .gitattributes +5 -0
  2. PACKAGING_NOTES.md +1 -0
  3. README.md +19 -0
  4. autoencoder.safetensors +3 -0
  5. config.json +21 -0
  6. mask.npz +3 -0
.gitattributes ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
2
+ *.pt filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.npz filter=lfs diff=lfs merge=lfs -text
5
+ *.model filter=lfs diff=lfs merge=lfs -text
PACKAGING_NOTES.md ADDED
@@ -0,0 +1 @@
 
 
1
+ - Skipped NeuroAdapter conversion: Can't get attribute 'InterleavedDecoderAdapter' on <module '__main__' from '/home/rph/neurovlm/scripts/prepare_hf_model_repos.py'>
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: neurovlm
3
+ tags:
4
+ - neuroimaging
5
+ - autoencoder
6
+ - brain-maps
7
+ ---
8
+
9
+ # NeuroAutoEncoder
10
+
11
+ Autoencoder weights and brain-mask metadata for NeuroVLM anatomical atlas vectors.
12
+
13
+ ## Files
14
+
15
+ - `autoencoder.safetensors`: `neurovlm.models.NeuroAutoEncoder` weights.
16
+ - `mask.npz`: mask array and affine used to map vectors back to NIfTI space.
17
+ - `config.json`: architecture and file metadata for downstream loaders.
18
+
19
+ Load through `neurovlm.models.load_model("autoencoder")` after the package loader update.
autoencoder.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d66a87175fe773607c8fa329793584c00c21983f09b13018d8cf106aa3ace5d
3
+ size 239712392
config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "adapter": {
3
+ "included": false,
4
+ "weights": null
5
+ },
6
+ "architectures": [
7
+ "neurovlm.models.NeuroAutoEncoder"
8
+ ],
9
+ "autoencoder": {
10
+ "dim_h0": 1024,
11
+ "dim_h1": 512,
12
+ "dim_latent": 384,
13
+ "dim_neuro": 28542,
14
+ "normalize_latent": false,
15
+ "out": "logit"
16
+ },
17
+ "library_name": "neurovlm",
18
+ "mask": "mask.npz",
19
+ "model_type": "neurovlm_autoencoder",
20
+ "weights": "autoencoder.safetensors"
21
+ }
mask.npz ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a0ea75fe77b8fc05de26f43efbe7549f540e1ff447400a4facf9a621c9ac9aa
3
+ size 2928