mranzinger commited on
Commit
aa1a477
1 Parent(s): 8e76cda

Add 'adaptors' property (#5)

Browse files

- Add 'adaptors' property (91107e1eeece9fc6717dc44bb8412bdc895410bf)

Files changed (1) hide show
  1. 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