code: def test_init_vit_weights(self):
      norm = torch.nn.LayerNorm(20)
      ones = torch.ones(20)
      zeros = torch.zeros(20)
      init_vit_weights(norm)
      self.assertTrue((norm.weight == ones).all())
      self.assertTrue((norm.bias == zeros).all())
  
imports: import unittest
  import torch
  from towhee.models.utils.init_vit_weights import init_vit_weights
label_desc: Use image node_id 1 for calling the InitVitWeightsTest obj's underlying member method code with example usage: obj.test_init_vit_weights() without return types