> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://docs.nvidia.com/nemo/automodel/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://docs.nvidia.com/nemo/automodel/_mcp/server.

# nemo_automodel.components.models.common.bidirectional

## Module Contents

### Classes

| Name                                                                                                        | Description                            |
| ----------------------------------------------------------------------------------------------------------- | -------------------------------------- |
| [`EncoderStateDictAdapter`](#nemo_automodel-components-models-common-bidirectional-EncoderStateDictAdapter) | Adapter for encoder model state dicts. |

### Data

[`__all__`](#nemo_automodel-components-models-common-bidirectional-__all__)

### API

```python
class nemo_automodel.components.models.common.bidirectional.EncoderStateDictAdapter()
```

**Bases:** [StateDictAdapter](/nemo-automodel/nemo_automodel/components/checkpoint/state_dict_adapter#nemo_automodel-components-checkpoint-state_dict_adapter-StateDictAdapter)

Adapter for encoder model state dicts.

Internal format uses a `model.` prefix on all keys.  HF format does not.
This adapter strips or adds the `model.` prefix as needed, including
for PEFT-wrapped keys (`base_model.model.model.X` \<-> `base_model.model.X`).

```python
nemo_automodel.components.models.common.bidirectional.EncoderStateDictAdapter._add_model_prefix(
    key
)
```

```python
nemo_automodel.components.models.common.bidirectional.EncoderStateDictAdapter._strip_model_prefix(
    key
)
```

```python
nemo_automodel.components.models.common.bidirectional.EncoderStateDictAdapter.convert_single_tensor_to_hf(
    fqn,
    tensor,
    kwargs = {}
)
```

```python
nemo_automodel.components.models.common.bidirectional.EncoderStateDictAdapter.from_hf(
    hf_state_dict,
    device_mesh = None,
    kwargs = {}
)
```

```python
nemo_automodel.components.models.common.bidirectional.EncoderStateDictAdapter.to_hf(
    state_dict,
    kwargs = {}
)
```

```python
nemo_automodel.components.models.common.bidirectional.__all__ = ['EncoderStateDictAdapter']
```