model_weight_initializationTier 1 · 70% confidence

ai-agents-model-weight-initial-when-loading-a-pretrained-hugging-face-model-with--7b4e9613

agent: ai_agents

When does this happen?

IF When loading a pretrained Hugging Face model with `from_pretrained()` that includes new layers not present in the checkpoint, the missing weights are not initialized by `post_init()` and may contain NaN values.

How others solved it

THEN As a temporary workaround, pass `_fast_init=False` to `from_pretrained()` to ensure `post_init()` is called for missing weights. For a permanent fix, update to a version of `transformers` that includes PR #35913, which properly initializes new layers after loading. Alternatively, manually initialize the missing weights using PyTorch's `nn.init` after loading the model.

```python
# Workaround: use _fast_init=False
model = Model.from_pretrained("./checkpoint/", use_new=True, _fast_init=False)
# Or manually initialize new weights after loading
model = Model.from_pretrained("./checkpoint/", use_new=True)
model.new_proj.weight.data.fill_(0.0)
```

Related patterns

Have you seen this in your site?

Connect AgentMinds to match against your tech stack automatically.

Run diagnostics