Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -86,8 +86,8 @@ This model performs **Contextualized Structured Radiology Report Generation (CSR
|
|
| 86 |
- Learning rate scheduler: Cosine with 3% warmup
|
| 87 |
- Precision: bfloat16
|
| 88 |
- Attention implementation: Flash Attention 2
|
| 89 |
-
- Max sequence length:
|
| 90 |
-
- Max images per sample:
|
| 91 |
|
| 92 |
**Hardware:**
|
| 93 |
- GPU: NVIDIA A100 (or equivalent)
|
|
@@ -113,7 +113,7 @@ model = AutoModelForVision2Seq.from_pretrained(
|
|
| 113 |
processor = AutoProcessor.from_pretrained("StanfordAIMI/CheXagent-2-3b", trust_remote_code=True)
|
| 114 |
|
| 115 |
# Load chest X-ray images (current and prior studies)
|
| 116 |
-
# CSRRG models support multiple images for temporal comparison (max_images_per_sample:
|
| 117 |
current_image = Image.open("current_xray.jpg")
|
| 118 |
prior_image = Image.open("prior_xray.jpg")
|
| 119 |
|
|
|
|
| 86 |
- Learning rate scheduler: Cosine with 3% warmup
|
| 87 |
- Precision: bfloat16
|
| 88 |
- Attention implementation: Flash Attention 2
|
| 89 |
+
- Max sequence length: 4096
|
| 90 |
+
- Max images per sample: 2
|
| 91 |
|
| 92 |
**Hardware:**
|
| 93 |
- GPU: NVIDIA A100 (or equivalent)
|
|
|
|
| 113 |
processor = AutoProcessor.from_pretrained("StanfordAIMI/CheXagent-2-3b", trust_remote_code=True)
|
| 114 |
|
| 115 |
# Load chest X-ray images (current and prior studies)
|
| 116 |
+
# CSRRG models support multiple images for temporal comparison (max_images_per_sample: 2)
|
| 117 |
current_image = Image.open("current_xray.jpg")
|
| 118 |
prior_image = Image.open("prior_xray.jpg")
|
| 119 |
|