Skip to content

Commit 9707027

Browse files
committed
docs(huggingface): small fix
1 parent cf3cba6 commit 9707027

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/modalities/conversion/gpt2/conversion_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,8 +131,8 @@ def _copy_weights_model(hf_model: GPT2ForCausalLM, modalities_model: GPT2LLM):
131131
"""Copies the weights of the modalities model to the Huggingface transformers model.
132132
133133
Args:
134-
hf_model_model (GPT2ForCausalLM): The uninitialized Huggingface transformers model.
135-
The weights will be copied here.
134+
hf_model (GPT2ForCausalLM): The uninitialized Huggingface transformers model.
135+
The weights will be copied here.
136136
modalities_model (GPT2LLM): The modalities model from which the weights will be copied.
137137
"""
138138
hf_model.model.embed_tokens.weight.data.copy_(modalities_model.transformer.wte.weight.data)

0 commit comments

Comments
 (0)