nvidia
/

C-RADIOv2-B

Image Feature Extraction

feature-extraction

Model card Files Files and versions

mranzinger commited on Jan 30

Commit

cb109ca

·

verified ·

1 Parent(s): f6c0c44

Update hf_model.py

Files changed (1) hide show

hf_model.py +2 -0

hf_model.py CHANGED Viewed

@@ -97,6 +97,8 @@ class RADIOModel(PreTrainedModel):
     def __init__(self, config: RADIOConfig):
         super().__init__(config)
         RADIOArgs = namedtuple("RADIOArgs", config.args.keys())
         args = RADIOArgs(**config.args)

     def __init__(self, config: RADIOConfig):
         super().__init__(config)
+        if hasattr(super(), "post_init"):
+            super().post_init()
         RADIOArgs = namedtuple("RADIOArgs", config.args.keys())
         args = RADIOArgs(**config.args)