Shift import

This commit is contained in:
Vik Paruchuri
2025-11-19 12:18:57 -05:00
parent b96eb84094
commit c049e7524f

View File

@@ -1,7 +1,5 @@
from typing import List
from qwen_vl_utils import process_vision_info
from chandra.model.schema import BatchInputItem, GenerationResult
from chandra.model.util import scale_to_fit
from chandra.prompts import PROMPT_MAPPING
@@ -15,6 +13,8 @@ def generate_hf(
bbox_scale: int = settings.BBOX_SCALE,
**kwargs,
) -> List[GenerationResult]:
from qwen_vl_utils import process_vision_info
if max_output_tokens is None:
max_output_tokens = settings.MAX_OUTPUT_TOKENS