upload stage-3 model

README.md

@@ -58,7 +58,7 @@ The following results are reported on the RadGenome-Brain MRI evaluation split i
 
 ### MRI diagnosis and report generation
 
-| Available modalities | Top-1 diagnosis accuracy (%) | ROUGE |
+| Available modalities | Acc | ROUGE |
 | --- | ---: | ---: |
 | T1w only | 74.47 | 36.93 |
 | T1w + T2w | 76.60 | 38.23 |
@@ -67,7 +67,7 @@ The following results are reported on the RadGenome-Brain MRI evaluation split i
 
 ### MRI modality imputation
 
-| Imputation sequence | PSNR | Downstream Top-1 accuracy (%) |
+| Imputation sequence | PSNR | Acc |
 | --- | ---: | ---: |
 | T1w → T2w | 22.23 | 68.09 |
 | T1w, T2w → T2-FLAIR | 22.58 | 67.38 |

checkpoints_s3/0000200/ema.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6c5a61933959ccdcbb14c672ce208c87d3cd33441682e36a4177264dea8ee9f3
+size 29214685336

checkpoints_s3/0000200/llm_config.json

@@ -0,0 +1,27 @@
+{
+  "architectures": [
+    "Qwen2ForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 3584,
+  "initializer_range": 0.02,
+  "intermediate_size": 18944,
+  "max_position_embeddings": 32768,
+  "max_window_layers": 28,
+  "model_type": "qwen2",
+  "num_attention_heads": 28,
+  "num_hidden_layers": 28,
+  "num_key_value_heads": 4,
+  "rms_norm_eps": 1e-06,
+  "rope_theta": 1000000.0,
+  "sliding_window": 131072,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.43.1",
+  "use_cache": true,
+  "use_sliding_window": false,
+  "vocab_size": 152064
+}

checkpoints_s3/0000200/tokenizer_config.json

@@ -0,0 +1 @@
+/volume/med-train/users/cliu05/UFM/code/UniBrain/models/BAGEL-7B-MoT/tokenizer_config.json

checkpoints_s3/0000200/vit_config.json

@@ -0,0 +1,9 @@
+{
+    "hidden_size": 1152,
+    "image_size": 980,
+    "intermediate_size": 4304,
+    "model_type": "siglip_vision_model",
+    "num_attention_heads": 16,
+    "num_hidden_layers": 27,
+    "patch_size": 14
+}