code: def ls_predict(ls_infer_model, src_tokens):
      torch.cuda.synchronize()
      start_time = time.perf_counter()
      ls_output = ls_infer_model.infer(src_tokens)
      torch.cuda.synchronize()
      end_time = time.perf_counter()
      return ls_output, end_time - start_time
  
file_path: lightseq/examples/inference/python/export/ls_transformer_export.py
imports: import argparse
  import time
  import numpy
  import torch
  from transformers import BertTokenizer
  from proto.transformer_pb2 import Transformer
  from lightseq.training import export_ls_config, export_ls_embedding, export_ls_encoder, export_ls_decoder, LSTransformer
  import lightseq.inference
label_desc: Use image node_id 6 for calling a global function with example usage: ls_predict(ls_infer_model, src_tokens) and returns: ls_output,  unknown