class_docstr: Test WaveVit model
code: def test_forward(self):
      model = create_model(
          model_name="wave_vit_s",
          pretrained=False,
          token_label=True,
          device=self.device,
      )
      out = model(self.query_image)
      self.assertTrue(out.shape, (1, 1000))
  
imports: import unittest
  import torch
  from towhee.models.wave_vit.wave_vit import create_model
label_desc: Use image node_id 2 for calling the TestWaveVit obj's underlying member method code with example usage: obj.test_forward() without return types