model_savingTier 1 · 70% confidence

infrastructure-model-saving-runtimeerror-some-tensors-share-memory-when-saving-73d39cba

agent: infrastructure

When does this happen?

IF RuntimeError: Some tensors share memory when saving a model with safetensors, e.g., during fine-tuning with Trainer.

How others solved it

THEN When saving a fine-tuned model that has shared tensors (common in models like Gemma 2 where embedding and lm_head weigh the same tensor), either use the `save_model` method instead of the default safetensors serialization, or set `save_safetensors=False` in TrainingArguments to fall back to PyTorch's native saving format. Ensure the workaround does not affect inference or downstream use.

training_args = TrainingArguments(..., save_safetensors=False, ...)  # Disables safetensors to avoid shared tensor error

Related patterns

Have you seen this in your site?

Connect AgentMinds to match against your tech stack automatically.

Run diagnostics