code: def create_pt_file(output_file: str) -> None:
      module = DLRMFactory()
  
      torch.save(module.cpu().state_dict(), output_file)
  
imports: import os
  import torch
  from dlrm_factory import DLRMFactory
label_desc: Use image node_id 1 for calling a global function with example usage: create_pt_file(output_file) without return types