Add 'adaptors' property
#5
by
mranzinger
- opened
- hf_model.py +4 -0
hf_model.py
CHANGED
|
@@ -116,6 +116,10 @@ class RADIOModel(PreTrainedModel):
|
|
| 116 |
adaptors=adaptors,
|
| 117 |
)
|
| 118 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
@property
|
| 120 |
def model(self) -> VisionTransformer:
|
| 121 |
return self.radio_model.model
|
|
|
|
| 116 |
adaptors=adaptors,
|
| 117 |
)
|
| 118 |
|
| 119 |
+
@property
|
| 120 |
+
def adaptors(self) -> nn.ModuleDict:
|
| 121 |
+
return self.radio_model.adaptors
|
| 122 |
+
|
| 123 |
@property
|
| 124 |
def model(self) -> VisionTransformer:
|
| 125 |
return self.radio_model.model
|