File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -11,7 +11,7 @@ def __init__(self):
1111 self .position_sin = None
1212
1313 def init_some_extra_state (self , model , input_ids : torch .Tensor ):
14- super ().init_some_extra_state (self , model , input_ids )
14+ super ().init_some_extra_state (model , input_ids )
1515 if self .is_prefill :
1616 b_ready_cache_len_numpy = self .b_ready_cache_len .cpu ().numpy ()
1717 self .b_ready_cache_len_numpy = b_ready_cache_len_numpy
Original file line number Diff line number Diff line change 1515
1616
1717def seed_all (seed ):
18+ if not seed :
19+ seed = int (time .time ())
1820 random .seed (seed )
1921 os .environ ["PYTHONHASHSEED" ] = str (seed )
2022 np .random .seed (seed )
@@ -297,7 +299,7 @@ def main():
297299 parser .add_argument ("--server_api" , type = str , default = "lightllm" )
298300 parser .add_argument ("--dump_file" , type = str , default = "" )
299301 parser .add_argument ("--seed" , type = int , default = 0 )
300- parser .add_argument ("--range_ratio" , type = float , default = 0.5 )
302+ parser .add_argument ("--range_ratio" , type = float , default = 1.0 )
301303 parser .add_argument (
302304 "--force_terminate" ,
303305 type = int ,
You can’t perform that action at this time.
0 commit comments