Check if a flat layer index is the last layer of its stack.
(&self, flat_idx: usize)
| 119 | |
| 120 | /// Check if a flat layer index is the last layer of its stack. |
| 121 | pub fn is_stack_last(&self, flat_idx: usize) -> bool { |
| 122 | let (stack_idx, layer_in_stack) = self.flat_to_stack(flat_idx); |
| 123 | layer_in_stack == self.model.fm_decoder_num_layers[stack_idx] - 1 |
| 124 | } |
| 125 | } |
| 126 | |
| 127 | fn default_ff_dim() -> usize { 1536 } |
nothing calls this directly
no test coverage detected