Skip to content

Commit e81a8c7

Browse files
committed
[fix] minor bug - for image-based remote inference pipeline
1 parent 4c3062e commit e81a8c7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

compressai_vision/pipelines/remote_inference/image_remote_inference.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,8 +148,8 @@ def __call__(
148148
timing["decode"].append((end - start))
149149

150150
start = time_measure()
151-
# dec_d = {"file_name": dec_seq["file_names"][0]}
152-
dec_d = {"file_name": dec_seq[0]["file_names"][0]}
151+
dec_d = {"file_name": dec_seq["file_names"][0]}
152+
# dec_d = {"file_name": dec_seq[0]["file_names"][0]}
153153

154154
pred = vision_model.forward(org_map_func(dec_d))
155155
end = time_measure()

0 commit comments

Comments
 (0)