> This page is for version 0.7.0 · 26.10.
> For other versions, use one of these documentation indexes:
> - Nightly (default): https://docs.nvidia.com/nemo/automodel/nightly/llms.txt
> - Latest: https://docs.nvidia.com/nemo/automodel/latest/llms.txt
> - 0.7.0 · 26.10: https://docs.nvidia.com/nemo/automodel/v0.7/llms.txt
> - 0.5.0 · 26.06: https://docs.nvidia.com/nemo/automodel/v0.5/llms.txt
> - 0.4.0 · 26.04: https://docs.nvidia.com/nemo/automodel/v0.4/llms.txt

> 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.

# nemo_automodel.components.models.praxis.hub

The public weights by their Hugging Face repos: the checkpoint and the frozen encoders of the reference package.

## Module Contents

### Functions

| Name                                                                                              | Description                                                                                      |
| ------------------------------------------------------------------------------------------------- | ------------------------------------------------------------------------------------------------ |
| [`load_checkpoint_tensors`](#nemo_automodel-components-models-praxis-hub-load_checkpoint_tensors) | Read every shard that `model.safetensors.index.json` under `root` names.                         |
| [`load_text_encoder`](#nemo_automodel-components-models-praxis-hub-load_text_encoder)             | The Qwen3.5 text encoder of the reference package.                                               |
| [`load_vae`](#nemo_automodel-components-models-praxis-hub-load_vae)                               | The Wan2.2 VAE of the reference package.                                                         |
| [`resolve_repo`](#nemo_automodel-components-models-praxis-hub-resolve_repo)                       | Local path of `path`: itself when it exists, else the Hugging Face repo `path` in the Hub cache. |

### Data

[`WAN_VAE_FILENAME`](#nemo_automodel-components-models-praxis-hub-WAN_VAE_FILENAME)

### API

```python
nemo_automodel.components.models.praxis.hub.load_checkpoint_tensors(
    root: str | pathlib.Path,
    device: str | torch.device = 'cpu'
) -> dict[str, torch.Tensor]
```

Read every shard that `model.safetensors.index.json` under `root` names.

**Parameters:**

**`root`** `str | Path`

Checkpoint directory.

---

**`device`** `str | torch.device` — default: 'cpu'

Device of the loaded tensors.

---

**Returns:** `dict[str, Tensor]`

The checkpoint tensors under their checkpoint keys.

```python
nemo_automodel.components.models.praxis.hub.load_text_encoder(
    path: str | pathlib.Path,
    device: str | torch.device = 'cuda'
) -> praxis.text_encoder.TextEncoder
```

The Qwen3.5 text encoder of the reference package.

**Parameters:**

**`path`** `str | Path`

Directory in the Hugging Face layout, or the repo id (`Qwen/Qwen3.5-2B`).

---

**`device`** `str | torch.device` — default: 'cuda'

Device of the loaded model.

---

```python
nemo_automodel.components.models.praxis.hub.load_vae(
    path: str | pathlib.Path,
    device: str | torch.device = 'cuda',
    filename: str = WAN_VAE_FILENAME
) -> praxis.vae.Autoencoder
```

The Wan2.2 VAE of the reference package.

**Parameters:**

**`path`** `str | Path`

The `.pth` weights file, or the Hugging Face repo that holds it (`Wan-AI/Wan2.2-TI2V-5B`).

---

**`device`** `str | torch.device` — default: 'cuda'

Device of the loaded model.

---

**`filename`** `str` — default: WAN_VAE_FILENAME

Name of the weights file inside the repo.

---

```python
nemo_automodel.components.models.praxis.hub.resolve_repo(
    path: str | pathlib.Path,
    filename: str | None = None
) -> pathlib.Path
```

Local path of `path`: itself when it exists, else the Hugging Face repo `path` in the Hub cache.

**Parameters:**

**`path`** `str | Path`

Local file or directory, or a Hugging Face repo id such as `runwayml/praxis-1-droid`.

---

**`filename`** `str | None` — default: None

With a repo id, the single file to download instead of the whole repo.

---

**Returns:** `Path`

The local file or directory.

```python
nemo_automodel.components.models.praxis.hub.WAN_VAE_FILENAME = 'Wan2.2_VAE.pth'
```