Skip to content

Commit

Permalink
Fix types
Browse files Browse the repository at this point in the history
  • Loading branch information
pseudotensor committed Sep 14, 2024
1 parent a9b9715 commit 0e31621
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions openai_server/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ class H2oGPTParams(BaseModel):
image_audio_loaders: List | None = None
pdf_loaders: List | None = None
url_loaders: List | None = None
jq_schema: List | None = None
jq_schema: str | None = None
extract_frames: int | None = 10
llava_prompt: str | None = 'auto'
# visible_models
Expand All @@ -118,14 +118,14 @@ class H2oGPTParams(BaseModel):
hyde_template: str | None = 'auto'
hyde_show_only_final: bool | None = False
doc_json_mode: bool | None = False
metadata_in_context: str | None = 'auto'
metadata_in_context: Union[str, list] | None = 'auto'

chatbot_role: str | None = 'None'
speaker: str | None = 'None'
tts_language: str | None = 'autodetect'
tts_speed: float | None = 1.0

image_file: str | None = None
image_file: Union[str, list] | None = None
image_control: str | None = None
images_num_max: int | None = None
image_resolution: tuple | None = None
Expand All @@ -135,7 +135,7 @@ class H2oGPTParams(BaseModel):
image_batch_image_prompt: str | None = None
image_batch_final_prompt: str | None = None
image_batch_stream: bool | None = None
visible_vision_models: Union[str, int] | None = None
visible_vision_models: Union[str, int, list] | None = 'auto'
video_file: Union[str, list] | None = None

model_lock: dict | None = None
Expand Down
2 changes: 1 addition & 1 deletion src/version.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "6de12f281bbe2c85c3f3b988431b1e28c9ae4382"
__version__ = "a9b97158d1d5f3a8fbb47246e28327e7609bc903"

0 comments on commit 0e31621

Please sign in to comment.