huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-12 04:44:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source 
Handle NoneType in sliding_window comparison to fix TypeError in __init__.py by ensuring the comparison logic accounts for NoneType values, preventing errors and improving code robustness.
This commit is contained in:
Praz 2024-08-08 13:50:02 +05:30 committed by GitHub
parent bd4b23d0ba
commit d7c5ef6cd2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/text_generation_server/models/__init__.py
View File

@ -492,13 +492,7 @@ def get_model(
)
sliding_window = config_dict.get("sliding_window", -1)
if max_input_tokens is None:
max_input_tokens = 1024 # Set a default value if not provided
if sliding_window is None:
sliding_window = 2048 # Set a default value if not provided
if max_input_tokens <= sliding_window:
if max_input_tokens is not None and (sliding_window is None or max_input_tokens <= sliding_window):
sliding_window = -1
if (