aehrc
/

cxrmate-rrg24

Feature Extraction

vision-encoder-decoder

Model card Files Files and versions

anicolson commited on Jan 23, 2025

Commit

c1b861c

·

verified ·

1 Parent(s): 81c38cf

Update modelling_cxrrg.py

Files changed (1) hide show

modelling_cxrrg.py +2 -2

modelling_cxrrg.py CHANGED Viewed

@@ -504,7 +504,7 @@ class CXRRGModel(VisionEncoderDecoderModel):
                     report_seq_len,
                     report_seq_len,
                 ),
-                dtype=torch.long,
                 device=causal_2d_attention_mask.device,
             ),
         )
@@ -520,7 +520,7 @@ class CXRRGModel(VisionEncoderDecoderModel):
             1,
             prompt_seq_len,
             report_seq_len,
-            dtype=torch.long,
             device=causal_2d_attention_mask.device,
         )

                     report_seq_len,
                     report_seq_len,
                 ),
+                dtype=torch.float,
                 device=causal_2d_attention_mask.device,
             ),
         )
             1,
             prompt_seq_len,
             report_seq_len,
+            dtype=torch.float,
             device=causal_2d_attention_mask.device,
         )