Skip to content

Commit bf1046d

Browse files
author
none
committed
fix
1 parent 502c4c4 commit bf1046d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lightllm/common/basemodel/basemodel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -751,7 +751,7 @@ def _autotune_warmup(self):
751751
try:
752752
logger.info(f"autotune warmup for length {input_len}")
753753
rand_gen = torch.Generator(device="cuda")
754-
rand_gen.manual_seed(seed=input_len)
754+
rand_gen.manual_seed(input_len)
755755
dummy_input_ids = torch.randint(
756756
0, 10000, (input_len,), dtype=torch.int32, device="cuda", generator=rand_gen
757757
)

0 commit comments

Comments
 (0)