Fix torch.zeroes error

This commit is contained in:
doctorpangloss 2025-02-06 09:00:10 -08:00
parent 3f1f427ff4
commit 5b3eb2e51c

View File

@ -356,7 +356,7 @@ class TransformersManagedModel(ModelManageable, LanguageModel):
image_sizes = []
# todo: what is the best choice for this?
# probably select a size that related to the vision tower?
images = torch.zeroes((0, 0, 0, 3))
images = torch.zeros((0, 0, 0, 3))
try:
if hasattr(tokenizer, "apply_chat_template"):