class_docstr: Test HorNet model
code: def test_model_names(self):
      try:
          create_model(model_name="wrong name")
      except ValueError:
          pass
  
      hornet_tiny = create_model(
          model_name="hornet_tiny_7x7", device=self.device
      )
      self.assertTrue(hornet_tiny.base_dim == 64)
      self.assertTrue(hornet_tiny.num_classes == 1000)
  
      hornet_small = create_model(
          model_name="hornet_small_gf", device=self.device
      )
      self.assertTrue(hornet_small.base_dim == 96)
      self.assertTrue(hornet_small.num_classes == 1000)
  
      hornet_base = create_model(
          model_name="hornet_base_7x7", device=self.device
      )
      self.assertTrue(hornet_base.base_dim == 128)
      self.assertTrue(hornet_base.num_classes == 1000)
  
      hornet_large = create_model(
          model_name="hornet_large_gf_img384_22k", device=self.device
      )
      self.assertTrue(hornet_large.base_dim == 192)
      self.assertTrue(hornet_large.num_classes == 21841)
  
imports: import unittest
  import torch
  from towhee.models.hornet import create_model
label_desc: Use image node_id 2 for calling the TestModel obj's underlying member method code with example usage: obj.test_model_names() without return types