Skip to content

Commit c774dd0

Browse files
author
Wei Chu
committed
fix sanity
1 parent d869551 commit c774dd0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker_pytorch_serving_container/default_pytorch_inference_handler.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def default_predict_fn(self, data, model, context=None):
132132
else:
133133
if context:
134134
properties = context.system_properties
135-
device = torch.device("cuda:" + str(properties.get("gpu_id"))
135+
device = torch.device("cuda:" + str(properties.get("gpu_id"))
136136
if torch.cuda.is_available() else "cpu")
137137
else:
138138
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

0 commit comments

Comments
 (0)