nvidia
/

RADIO

Image Feature Extraction

feature-extraction

Model card Files Files and versions

gheinrich commited on Dec 11, 2023

Commit

c1fddb0

·

1 Parent(s): db40549

Upload model

Files changed (1) hide show

hf_model.py +2 -0

hf_model.py CHANGED Viewed

@@ -45,6 +45,8 @@ class RADIOConfig(PretrainedConfig):
 class RADIOModel(PreTrainedModel):
     """Pretrained Hugging Face model for RADIO."""
     def __init__(self, config):
         super().__init__(config)

 class RADIOModel(PreTrainedModel):
     """Pretrained Hugging Face model for RADIO."""
+    config_class = RADIOConfig
     def __init__(self, config):
         super().__init__(config)