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-01-05 18:33:11 -06:00
parent 75f9a6ab93
commit 4b2d6f5595
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
vall_e/models/base.py
View File

@ -951,16 +951,16 @@ class Base(nn.Module):
quant_levels: int | list[int] | Tensor | None = None quant_levels: int | list[int] | Tensor | None = None
): ):
if text_list: if text_list and text_list[0] is not None:
device = text_list[0].device device = text_list[0].device
batch_size = len(text_list) batch_size = len(text_list)
elif raw_text_list: elif raw_text_list and raw_text_list[0] is not None:
device = raw_text_list[0].device device = raw_text_list[0].device
batch_size = len(raw_text_list) batch_size = len(raw_text_list)
elif proms_list: elif proms_list and proms_list[0] is not None:
device = proms_list[0].device device = proms_list[0].device
batch_size = len(proms_list) batch_size = len(proms_list)
elif resps_list: elif resps_list and resps_list[0] is not None:
device = resps_list[0].device device = resps_list[0].device
batch_size = len(resps_list) batch_size = len(resps_list)