Upload model

README.md

@@ -1,9 +1,9 @@
 ---
-library_name: transformers
-license: apache-2.0
 datasets:
 - StanfordAIMI/interpret-cxr-test-public
 - StanfordAIMI/interpret-cxr-test-hidden
+library_name: transformers
+license: apache-2.0
 ---
 
 # CXRMate-RRG24: Entropy-Augmented Self-Critical Sequence Training for Radiology Report Generation

config.json

@@ -217,5 +217,5 @@
   "model_type": "vision-encoder-decoder",
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.0"
+  "transformers_version": "4.40.2"
 }

generation_config.json

@@ -3,5 +3,5 @@
   "bos_token_id": 1,
   "eos_token_id": 2,
   "pad_token_id": 4,
-  "transformers_version": "4.39.0"
+  "transformers_version": "4.40.2"
 }

modelling_cxrrg.py

@@ -7,9 +7,8 @@ from transformers import PreTrainedTokenizerFast, VisionEncoderDecoderModel
 from transformers.configuration_utils import PretrainedConfig
 from transformers.modeling_outputs import Seq2SeqLMOutput
 from transformers.modeling_utils import PreTrainedModel
-from transformers.models.vision_encoder_decoder.configuration_vision_encoder_decoder import (
-    VisionEncoderDecoderConfig,
-)
+from transformers.models.vision_encoder_decoder.configuration_vision_encoder_decoder import \
+    VisionEncoderDecoderConfig
 from transformers.utils import logging
 
 from .modelling_uniformer import MultiUniFormerWithProjectionHead

modelling_uniformer.py

@@ -1,7 +1,7 @@
 from collections import OrderedDict
 from functools import partial
-from typing import Optional, Tuple, Union
 from math import isqrt
+from typing import Optional, Tuple, Union
 
 import torch
 import torch.nn as nn