huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-01 15:02:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleanup prints

Browse Source
This commit is contained in:
Miquel Farre 2024-12-11 21:09:17 +00:00 committed by drbh
parent 19e1c8da31
commit db97d979fb
1 changed files with 0 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
server/text_generation_server/models/vlm_causal_lm.py
View File

@ -244,21 +244,6 @@ class VlmCausalLMBatch(FlashCausalLMBatch):
) )
num_bytes = len(video_frame_buf) num_bytes = len(video_frame_buf)
bytes_per_frame = num_bytes // chunk.video.frames bytes_per_frame = num_bytes // chunk.video.frames
#height = bytes_per_frame // 3 // chunk.video.width
from loguru import logger
log_master(
logger.info,
f"Video buffer size: {len(video_frame_buf)}",
)
log_master(
logger.info,
f"Frames in chunk: {chunk.video.frames}",
)
log_master(
logger.info,
f"Bytes per frame: {bytes_per_frame}",
)
# iterate over with a stride the size of a frame # iterate over with a stride the size of a frame
frames = [] frames = []