Add @dataclass decorator to CXRBertOutput

#6

Should fix https://github.com/huggingface/transformers/issues/30412

(Someone logged this bug on the huggingface/transformer's repo as opposed to microsoft/hi-ml.)

The file uploaded is the main version of: https://github.com/microsoft/hi-ml/blame/main/hi-ml-multimodal/src/health_multimodal/text/model/modelling_cxrbert.py

The file uploaded should not be merged, it's a reference. Still figuring out how HF's PR's work.

Microsoft org

Thanks, @kendallpark . Closing in favour of #7.

fepegar changed pull request status to closed

Sign up or log in to comment