Commit2d7af4b

authored

[https://nvbugs/5468897][fix] fix invalid expression for disabling pa… (#7762)

Signed-off-by: nv-guomingz <137257613+nv-guomingz@users.noreply.github.com>

1 parentb940ebf commit2d7af4bCopy full SHA for 2d7af4b

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -831,7 +831,7 @@ def load_single_module(name, module):`
`831`	`831`	`p.data.copy_(module_weights[n][:])`
`832`	`832`
`833`	`833`	`ifos.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"]:`
`835`	`835`	`forname,moduleintqdm(list(model.named_modules()),`
`836`	`836`	`desc="Loading weights"):`
`837`	`837`	`load_single_module(name,module)`

Comments

(0)