Fixed style issue

cehongwang · cehongwang · commit 12d1d80a5df3 · 2024-08-21T12:06:35.000-07:00
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
@@ -534,4 +534,4 @@ def save(
                     exp_program = torch.export.export(
                         module, tuple(arg_inputs), kwargs=kwarg_inputs, strict=False
                     )
-                    torch.export.save(exp_program, file_path)
+                    torch.export.save(exp_program, file_path)
diff --git a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py
@@ -361,7 +361,7 @@ def check_weight_equal(
             network_weight = torch.from_numpy(network_weight).cuda()
         try:
             return sd_weight.shape == network_weight.shape and torch.all(
-                torch.abs(sd_weight - network_weight) < 0.1
+                torch.abs(sd_weight - network_weight) < 0.01
             )
         except Exception:
             return torch.all(sd_weight == network_weight)

Original file line number	Diff line number	Diff line change
`@@ -534,4 +534,4 @@ def save(`
`534`	`534`	`exp_program = torch.export.export(`
`535`	`535`	`module, tuple(arg_inputs), kwargs=kwarg_inputs, strict=False`
`536`	`536`	`)`
`537`		`- torch.export.save(exp_program, file_path)`
	`537`	`+ torch.export.save(exp_program, file_path)`
Original file line number	Diff line number	Diff line change
`@@ -361,7 +361,7 @@ def check_weight_equal(`
`361`	`361`	`network_weight = torch.from_numpy(network_weight).cuda()`
`362`	`362`	`try:`
`363`	`363`	`return sd_weight.shape == network_weight.shape and torch.all(`
`364`		`- torch.abs(sd_weight - network_weight) < 0.1`
	`364`	`+ torch.abs(sd_weight - network_weight) < 0.01`
`365`	`365`	`)`
`366`	`366`	`except Exception:`
`367`	`367`	`return torch.all(sd_weight == network_weight)`