mirror of
https://github.com/allenai/olmocr.git
synced 2025-11-16 02:17:52 +00:00
Fixing other way to run benchmark
This commit is contained in:
parent
eaf83026d3
commit
255ee48594
@ -22,6 +22,7 @@ logger = logging.getLogger("olmocr_runner")
|
|||||||
@dataclass
|
@dataclass
|
||||||
class Args:
|
class Args:
|
||||||
model: str = "allenai/olmOCR-7B-1025-FP8"
|
model: str = "allenai/olmOCR-7B-1025-FP8"
|
||||||
|
port: int = 3000
|
||||||
model_chat_template: str = "qwen2-vl"
|
model_chat_template: str = "qwen2-vl"
|
||||||
model_max_context: int = 8192
|
model_max_context: int = 8192
|
||||||
target_longest_image_dim: int = 1024
|
target_longest_image_dim: int = 1024
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user