Upload model
Browse files- modelling_cxrmate_ed.py +0 -1
modelling_cxrmate_ed.py
CHANGED
|
@@ -147,7 +147,6 @@ class CXRMateEDModel(VisionEncoderDecoderModel):
|
|
| 147 |
|
| 148 |
# Decoder:
|
| 149 |
if decoder is None:
|
| 150 |
-
assert not config.decoder.add_cross_attention
|
| 151 |
decoder = transformers.LlamaForCausalLM(config=config.decoder)
|
| 152 |
|
| 153 |
self.encoder = CXRStudyImagesEncoder(encoder, self.config.decoder)
|
|
|
|
| 147 |
|
| 148 |
# Decoder:
|
| 149 |
if decoder is None:
|
|
|
|
| 150 |
decoder = transformers.LlamaForCausalLM(config=config.decoder)
|
| 151 |
|
| 152 |
self.encoder = CXRStudyImagesEncoder(encoder, self.config.decoder)
|