code: def get_output_model_version(script_module: torch.nn.Module) -> int:
      buffer = io.BytesIO()
      torch.jit.save(script_module, buffer)
      buffer.seek(0)
      zipped_model = zipfile.ZipFile(buffer)
      try:
          version = int(
              zipped_model.read("archive/version").decode("utf-8")
          )
          return version
      except KeyError:
          version = int(
              zipped_model.read("archive/.data/version").decode("utf-8")
          )
          return version
  
file_path: pytorch/test/jit/fixtures_srcs/generate_models.py
imports: import io
  import logging
  import sys
  import zipfile
  from pathlib import Path
  from typing import Set
  import torch
  from test.jit.fixtures_srcs.fixtures_src import *
  from torch.jit.mobile import _load_for_lite_interpreter, _export_operator_list
label_desc: Use image node_id 5 for calling a global function with example usage: get_output_model_version(script_module) and returns: version, version