Skip to content

Commit 2798bc7

Browse files
author
niushengxiao
committed
fix: fix bugs
1 parent e037c88 commit 2798bc7

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

lightllm/models/llama/infer_struct.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

test/benchmark_qps.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@
1515

1616

1717
def 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,

0 commit comments

Comments
 (0)