(&self, idx: usize)
| 88 | |
| 89 | impl ExpertProvider for StackedResidentProvider { |
| 90 | fn get_expert(&self, idx: usize) -> Result<ExpertWeights> { |
| 91 | Ok(ExpertWeights { |
| 92 | gate_proj: self.gate_proj.get(idx)?, |
| 93 | up_proj: self.up_proj.get(idx)?, |
| 94 | down_proj: self.down_proj.get(idx)?, |
| 95 | }) |
| 96 | } |
| 97 | |
| 98 | fn num_experts(&self) -> usize { |
| 99 | self.num_experts |