Upload model
Browse files- hf_model.py +2 -0
hf_model.py
CHANGED
|
@@ -45,6 +45,8 @@ class RADIOConfig(PretrainedConfig):
|
|
| 45 |
class RADIOModel(PreTrainedModel):
|
| 46 |
"""Pretrained Hugging Face model for RADIO."""
|
| 47 |
|
|
|
|
|
|
|
| 48 |
def __init__(self, config):
|
| 49 |
super().__init__(config)
|
| 50 |
|
|
|
|
| 45 |
class RADIOModel(PreTrainedModel):
|
| 46 |
"""Pretrained Hugging Face model for RADIO."""
|
| 47 |
|
| 48 |
+
config_class = RADIOConfig
|
| 49 |
+
|
| 50 |
def __init__(self, config):
|
| 51 |
super().__init__(config)
|
| 52 |
|