code: def __init__(self, poly_config: PolyConfig):
      super().__init__()
  
      self.poly_type = poly_config.poly_type
      self.n_tasks = poly_config.n_tasks
      self.n_skills = poly_config.n_skills
      self.n_splits = poly_config.n_splits
  
      self.module_logits = nn.Parameter(
          torch.empty((self.n_tasks, self.n_splits * self.n_skills))
      )
  
file_path: peft/src/peft/tuners/poly/router.py
imports: from abc import ABC, abstractmethod
  import torch
  from torch import nn
  from torch.distributions.relaxed_bernoulli import RelaxedBernoulli
  from .config import PolyConfig
label_desc: Use image node_id 1 to create a new PolyRouter object from inherited base classes: Router with example: obj = PolyRouter(poly_config)