code: def __init__(self, config, prefix, weights):
      super().__init__()
      self.linear = FastLinear.load(
          config, prefix=f"{prefix}.linear", weights=weights, bias=True
      )
      self.act = torch.nn.SiLU()
  
file_path: text-generation-inference/server/text_generation_server/utils/medusa.py
imports: import torch
  from dataclasses import dataclass
  from text_generation_server.utils.layers import TensorParallelHead, FastLinear
label_desc: Use image node_id 1 to create a new ResBlock object from inherited base classes: torch.nn with example: obj = ResBlock(config, prefix, weights)