Added a missing argument

France-Travail · Jul 3, 2024 · f208adb · f208adb
1 parent 9b0830f
commit f208adb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/happy_vllm/utils_args.py b/src/happy_vllm/utils_args.py
@@ -156,6 +156,7 @@ class ModelSettings(BaseSettings):
         tokenizer_pool_type: str = default_args.tokenizer_pool_type
         tokenizer_pool_extra_config: Optional[str] = default_args.tokenizer_pool_extra_config
         image_input_type: Optional[str] = default_args.image_input_type
+        image_token_id: Optional[int] = None
         image_input_shape: Optional[str] = default_args.image_input_shape
         image_feature_size: Optional[int] = default_args.image_feature_size
         image_processor: Optional[str] = None