biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Misc][Build] Lazy load cv2 in nemotron_parse.py (#33189)

Browse Source 
Signed-off-by: kiersten-stokes <kierstenstokes@gmail.com>
This commit is contained in:
Kiersten Stokes
2026-01-29 00:55:50 -06:00
committed by GitHub
parent f9d03599ef
commit 9e138cb01d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
vllm/model_executor/models/nemotron_parse.py
View File

@@ -11,7 +11,6 @@ import math
from collections.abc import Iterable, Mapping, Sequence from collections.abc import Iterable, Mapping, Sequence
from typing import Annotated, Literal from typing import Annotated, Literal
import cv2
import numpy as np import numpy as np
import torch import torch
import torch.nn as nn import torch.nn as nn
@@ -416,6 +415,8 @@ class NemotronParseImageProcessor:
else: else:
self.target_height = self.target_width = int(self.final_size) self.target_height = self.target_width = int(self.final_size)
import cv2
self.transform = A.Compose( self.transform = A.Compose(
[ [
A.PadIfNeeded( A.PadIfNeeded(
@@ -457,6 +458,8 @@ class NemotronParseImageProcessor:
new_height = int(new_width / aspect_ratio) new_height = int(new_width / aspect_ratio)
# Use cv2.INTER_LINEAR like the original # Use cv2.INTER_LINEAR like the original
import cv2
return cv2.resize( return cv2.resize(
image, (new_width, new_height), interpolation=cv2.INTER_LINEAR image, (new_width, new_height), interpolation=cv2.INTER_LINEAR
) )