Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions chandra/model/hf.py
Original file line number Diff line number Diff line change
@@ -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
Expand All @@ -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

Expand Down