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

Update __init__.py

Browse Source 
Fix issue with NoneType comparison for max_input_tokens and sliding_window

- Add default values for max_input_tokens and sliding_window to handle None cases.
- Ensure the comparison between max_input_tokens and sliding_window is handled correctly to prevent TypeError.
- This change addresses the error: TypeError: '<=' not supported between instances of 'int' and 'NoneType'.
This commit is contained in:
Praz 2024-08-08 13:43:01 +05:30 committed by GitHub
parent a379d5536b
commit bd4b23d0ba
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 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,7 +492,13 @@ def get_model(
)
sliding_window = config_dict.get("sliding_window", -1)
if max_input_tokens is not None and max_input_tokens <= sliding_window:
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:
sliding_window = -1
if (