↓ 5 callersMethodforward(self, outputs, cls_output=None, attention_mask=None, return_dict=False, **kwargs)
src/adapters/heads/base.py:337
↓ 4 callersMethodadd_custom_head(self, head_type, head_name, overwrite_ok=False, set_active=True, **kwargs)
src/adapters/heads/model_mixin.py:302
↓ 3 callersMethodtrainings_run(
self,
model,
lr=1e-5,
steps=8,
batch_size=2,
gradient_accumu
tests/test_methods/method_test_impl/composition/test_multi_task.py:42
↓ 2 callersMethod__init__(
self,
model,
head_name,
vocab_size=None,
embedding_size=None,
src/adapters/heads/language_modeling.py:13