riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

fix: remove undefined variable (#39146)

Browse Source
This commit is contained in:
ybkurt 2025-07-01 20:10:29 +03:00 committed by GitHub
parent cdfe49a4d0
commit 260846efad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/musicgen/modeling_musicgen.py
View File

@ -147,7 +147,7 @@ class MusicgenSinusoidalPositionalEmbedding(nn.Module):
position_ids = (torch.arange(seq_len) + past_key_values_length).to(input_ids.device) position_ids = (torch.arange(seq_len) + past_key_values_length).to(input_ids.device)
# expand embeddings if needed # expand embeddings if needed
if seq_len > self.weights.size(0): if seq_len > self.weights.size(0):
self.make_weights(seq_len + self.offset, self.embedding_dim) self.make_weights(seq_len, self.embedding_dim)
return self.weights.index_select(0, position_ids.view(-1)).detach() return self.weights.index_select(0, position_ids.view(-1)).detach()

2
src/transformers/models/musicgen_melody/modeling_musicgen_melody.py
View File

@ -154,7 +154,7 @@ class MusicgenMelodySinusoidalPositionalEmbedding(nn.Module):
position_ids = (torch.arange(seq_len) + past_key_values_length).to(inputs_embeds.device) position_ids = (torch.arange(seq_len) + past_key_values_length).to(inputs_embeds.device)
# expand embeddings if needed # expand embeddings if needed
if seq_len > self.weights.size(0): if seq_len > self.weights.size(0):
self.make_weights(seq_len + self.offset, self.embedding_dim) self.make_weights(seq_len, self.embedding_dim)
return self.weights.index_select(0, position_ids.view(-1)).detach() return self.weights.index_select(0, position_ids.view(-1)).detach()