refactoring

modeling_radzero.py

@@ -9,7 +9,7 @@ from transformers.trainer import logger
 
 from .align_transformers import build_align_transformer
 from .common_layers import BasePreTrainedModel
-from .configuration import CxrAlignConfig
+from .configuration_radzero import CxrAlignConfig
 from .losses import KeyPhraseAlignmentLoss
 from .text_encoders import aggregate_tokens, build_text_encoder
 from .vision_encoders import MRM, Dinov2Model, build_vision_encoder