code: def enable_streaming_llm(
      model, start_size, recent_size, use_flash_attn=True
  ):
      if "llama" in model.config.model_type:
          k_seq_dim = v_seq_dim = 2
          from streaming_llm.pos_shift.modify_llama import (
              enable_llama_pos_shift_attention,
          )
  
          enable_llama_pos_shift_attention(model, use_flash_attn)
      elif "mpt" in model.config.model_type:
          v_seq_dim = 2
          k_seq_dim = 3
      elif "gpt_neox" in model.config.model_type:
          k_seq_dim = v_seq_dim = 2
          from streaming_llm.pos_shift.modify_gpt_neox import (
              enable_gpt_neox_pos_shift_attention,
          )
  
          enable_gpt_neox_pos_shift_attention(model)
      elif "falcon" in model.config.model_type:
          v_seq_dim = 1
          k_seq_dim = 1
          from streaming_llm.pos_shift.modify_falcon import (
              enable_falcon_pos_shift_attention,
          )
  
          enable_falcon_pos_shift_attention(model)
      else:
          raise ValueError(f"got {model.config.model_type}")
      kv_cache = StartRecentKVCache(
          start_size=start_size,
          recent_size=recent_size,
          k_seq_dim=k_seq_dim,
          v_seq_dim=v_seq_dim,
      )
      return kv_cache
  
label_desc: Use image node_id 1 for calling a global function with example usage: enable_streaming_llm(model, start_size, recent_size, use_flash_attn) and returns: kv_cache