LLM/chandra
1
0
Fork 0
mirror of https://github.com/datalab-to/chandra.git synced 2025-11-29 00:23:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #47 from datalab-to/dev

Browse Source 
Shift import
This commit is contained in:
Vik Paruchuri
2025-11-19 12:19:23 -05:00
committed by GitHub
parent 7ac08e16e1 c049e7524f
commit cba67c6d15
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
chandra/model/hf.py
View File

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