Skip to content
This repository was archived by the owner on Oct 9, 2024. It is now read-only.

Commit 655179f

Browse files
committed
fix broken generate in 4.26.1
1 parent a17b7d3 commit 655179f

File tree

2 files changed

+14
-14
lines changed

2 files changed

+14
-14
lines changed

.pre-commit-config.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
repos:
22
- repo: https://github.com/pycqa/isort
3-
rev: 5.10.1
3+
rev: 5.12.0
44
hooks:
55
- id: isort
66
name: isort (python)
77
- repo: https://github.com/psf/black
8-
rev: 22.8.0
8+
rev: 23.1.0
99
hooks:
1010
- id: black
1111
args: [--line-length=119,--target-version=py35]

inference_server/utils/requests.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -10,24 +10,24 @@ class BaseResponse(BaseModel):
1010

1111
class GenerateRequest(BaseModel):
1212
text: List[str] = None
13-
min_length: int = None
14-
do_sample: bool = None
15-
early_stopping: bool = None
16-
temperature: float = None
17-
top_k: int = None
18-
top_p: float = None
19-
typical_p: float = None
20-
repetition_penalty: float = None
13+
min_length: int = 0
14+
do_sample: bool = False
15+
early_stopping: bool = False
16+
temperature: float = 1
17+
top_k: int = 50
18+
top_p: float = 1
19+
typical_p: float = 1
20+
repetition_penalty: float = 1
2121
bos_token_id: int = None
2222
pad_token_id: int = None
2323
eos_token_id: int = None
24-
length_penalty: float = None
25-
no_repeat_ngram_size: int = None
26-
encoder_no_repeat_ngram_size: int = None
24+
length_penalty: float = 1
25+
no_repeat_ngram_size: int = 0
26+
encoder_no_repeat_ngram_size: int = 0
2727
max_time: float = None
2828
max_new_tokens: int = None
2929
decoder_start_token_id: int = None
30-
diversity_penalty: float = None
30+
diversity_penalty: float = 0
3131
forced_bos_token_id: int = None
3232
forced_eos_token_id: int = None
3333
exponential_decay_length_penalty: float = None

0 commit comments

Comments
 (0)