This repository was archived by the owner on Oct 9, 2024. It is now read-only.
File tree Expand file tree Collapse file tree 2 files changed +14
-14
lines changed Expand file tree Collapse file tree 2 files changed +14
-14
lines changed Original file line number Diff line number Diff line change 11repos :
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]
Original file line number Diff line number Diff line change @@ -10,24 +10,24 @@ class BaseResponse(BaseModel):
1010
1111class 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
You can’t perform that action at this time.
0 commit comments