HF_BaseModelWrapper

Same as `nn.Module`, but no need for subclasses to call `super().__init__`

HF_BaseModelWrapper(
  hf_model,
  output_hidden_states = FALSE,
  output_attentions = FALSE,
  ...
)

Arguments

hf_model

model

output_hidden_states

output hidden states

output_attentions

output attentions

...

additional arguments to pass

Value

None