From 0e91e131b8dd49375d31a3a14dc11f5ba667d3a9 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Mon, 22 Apr 2024 16:43:09 +0200 Subject: [PATCH] Update server/text_generation_server/models/custom_modeling/llava_next.py --- .../models/custom_modeling/llava_next.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/server/text_generation_server/models/custom_modeling/llava_next.py b/server/text_generation_server/models/custom_modeling/llava_next.py index 0f1944a7..14bf19e1 100644 --- a/server/text_generation_server/models/custom_modeling/llava_next.py +++ b/server/text_generation_server/models/custom_modeling/llava_next.py @@ -229,10 +229,6 @@ class LlavaNextForConditionalGeneration(nn.Module): self.config.image_grid_pinpoints, self.config.vision_config.image_size, ) - if image_sizes[image_idx][0].item() == 22: - import ipdb - - ipdb.set_trace() image_feature = image_feature.view( num_patch_height, num_patch_width, height, width, -1 )