Skip to content

Commit 2d7af4b

Browse files
authored
[https://nvbugs/5468897][fix] fix invalid expression for disabling pa… (#7762)
Signed-off-by: nv-guomingz <[email protected]>
1 parent b940ebf commit 2d7af4b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/models/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -831,7 +831,7 @@ def load_single_module(name, module):
831831
p.data.copy_(module_weights[n][:])
832832

833833
if os.environ.get("TRT_LLM_DISABLE_LOAD_WEIGHTS_IN_PARALLEL",
834-
True) in ["True", "true", "1", "yes", "y"]:
834+
"True") in ["True", "true", "1", "yes", "y"]:
835835
for name, module in tqdm(list(model.named_modules()),
836836
desc="Loading weights"):
837837
load_single_module(name, module)

0 commit comments

Comments
 (0)