mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

ugh

Browse Source
This commit is contained in:
mrq 2025-03-07 13:55:57 -06:00
parent 6afc2b7526
commit 89e52b9877
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/models/arch/llama.py
View File

@ -591,10 +591,6 @@ class Model(LlamaPreTrainedModel):
prom_start, prom_end = text_end, text_end + aux_len[1] prom_start, prom_end = text_end, text_end + aux_len[1]
output_start = prom_end output_start = prom_end
print( text_start, text_end )
print( prom_start, prom_end )
print( output_start )
expanded_mask[batch_index, 0, text_start:text_end, text_start:text_end] = 1.0 expanded_mask[batch_index, 0, text_start:text_end, text_start:text_end] = 1.0
expanded_mask[batch_index, 0, prom_start:prom_end, text_start:prom_end] = 1.0 expanded_mask[batch_index, 0, prom_start:prom_end, text_start:prom_end] = 1.0
expanded_mask[batch_index, 0, output_start:, :] = 1.0 expanded_mask[batch_index, 0, output_start:, :] = 1.0