code: def test_gather_16bit_params_on_model_save(gather_weights_key):
      config_dict = {
          gather_weights_key: True,
      }
      config = DeepSpeedZeroConfig(**config_dict)
  
      assert config.gather_16bit_weights_on_model_save == True
  
imports: import os
  import pytest
  import json
  import hjson
  import argparse
  from deepspeed.runtime.zero.config import DeepSpeedZeroConfig
  from deepspeed.accelerator import get_accelerator
  from unit.common import DistributedTest, get_test_path
  from unit.simple_model import SimpleModel, create_config_from_dict, random_dataloader
  import deepspeed.comm
  import deepspeed
  from deepspeed.runtime.config import DeepSpeedConfig, get_bfloat16_enabled
label_desc: Use image node_id 5 for calling a global function with example usage: test_gather_16bit_params_on_model_save(gather_weights_key) without return types