| 93 | layers_.push_back(LayerCache{ |
| 94 | std::move(keys[layer]), |
| 95 | std::move(values[layer]), |
| 96 | std::vector<float>(cache_elems, 0.0F), |
| 97 | std::vector<float>(cache_elems, 0.0F), |
| 98 | }); |
| 99 | } |
| 100 | } |
| 101 | |
| 102 | void TransformerKVCache::import_state(const TransformerKVState & state) { |
| 103 | current_end_ = state.current_end; |
| 104 | if (layers_.empty()) { |
| 105 | valid_steps_ = 0; |
| 106 | return; |
| 107 | } |
no outgoing calls
no test coverage detected