loadRoutedExpert method

RoutedExpertWeights loadRoutedExpert(
  1. int j, {
  2. bool keepFp16 = true,
})

Load one routed expert's SwiGLU triplet.

Implementation

RoutedExpertWeights loadRoutedExpert(int j, {bool keepFp16 = true}) =>
    RoutedExpertWeights(
      reader.readTensor(routedExpertGateKey(j), keepFp16: keepFp16),
      reader.readTensor(routedExpertUpKey(j), keepFp16: keepFp16),
      reader.readTensor(routedExpertDownKey(j), keepFp16: keepFp16),
    );