code: def test_pickling(mlp):
      test = ViT(mlp)
      _ = pickle.dumps(test)
      _ = deepcopy(test)
  
file_path: xformers/tests/test_pickling.py
imports: import pickle
  from copy import deepcopy
  import pytest
  from torch import nn
  from xformers import _is_triton_available
  from xformers.factory import xFormer, xFormerConfig
label_desc: Use image node_id 1 for calling a global function with example usage: test_pickling(mlp) without return types