mranzinger
commited on
Commit
•
aa1a477
1
Parent(s):
8e76cda
Add 'adaptors' property (#5)
Browse files- Add 'adaptors' property (91107e1eeece9fc6717dc44bb8412bdc895410bf)
- 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
|