Fixed the bug of SDXL Cuda Error

cehongwang · cehongwang · commit 05056fb860b8 · 2025-03-25T04:46:15.000Z
diff --git a/py/torch_tensorrt/dynamo/_refit.py b/py/torch_tensorrt/dynamo/_refit.py
@@ -123,7 +123,7 @@ def construct_refit_mapping_from_weight_name_map(
             engine_weight_map[engine_weight_name] = (
                 state_dict[sd_weight_name]
                 if state_dict[sd_weight_name].device == device
-                else state_dict[sd_weight_name].to("device")
+                else state_dict[sd_weight_name].to(device)
             )
 
         engine_weight_map[engine_weight_name] = (

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ def construct_refit_mapping_from_weight_name_map(`
`123`	`123`	`engine_weight_map[engine_weight_name] = (`
`124`	`124`	`state_dict[sd_weight_name]`
`125`	`125`	`if state_dict[sd_weight_name].device == device`
`126`		`- else state_dict[sd_weight_name].to("device")`
	`126`	`+ else state_dict[sd_weight_name].to(device)`
`127`	`127`	`)`
`128`	`128`
`129`	`129`	`engine_weight_map[engine_weight_name] = (`