Skip to content

Commit df2f746

Browse files
SangChengChiworldwzj
authored andcommitted
[FIX] new whisper
1 parent 5d40700 commit df2f746

File tree

3 files changed

+551
-3
lines changed

3 files changed

+551
-3
lines changed

lightllm/models/vit/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def _check_max_len_infer(self):
7070
).cuda()
7171
all_img_embeds = self.forward(dummy_images)
7272
del all_img_embeds
73-
logger.info(f"vit check max_len {self.batch_max_tokens} infer ok")
73+
logger.info(f"vit check max_len {self.max_batch_size} infer ok")
7474
except (RuntimeError, torch.OutOfMemoryError) as e:
7575
logger.exception(str(e))
7676
exception_str = (

0 commit comments

Comments
 (0)