update fil stage

nv-morpheus · Jul 15, 2024 · f0dbfc0 · f0dbfc0
1 parent 746530c
commit f0dbfc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/morpheus/stages/preprocess/preprocess_fil_stage.py b/morpheus/stages/preprocess/preprocess_fil_stage.py
@@ -115,7 +115,7 @@ def pre_process_batch(msg: ControlMessage, fea_len: int, fea_cols: typing.List[s
 
         # We need the C++ impl of TensorMemory until #1646 is resolved
         msg.tensors(_messages.TensorMemory(count=count, tensors={"input__0": data, "seq_ids": seg_ids}))
-        # msg.set_metadata("inference_memory_params", {"inference_type": "fil"})
+        msg.set_metadata("inference_memory_params", {"inference_type": "fil"})
         return msg
 
     def _get_preprocess_fn(