parent
5d6688de08
commit
c610b6c11b
9 changed files with 29 additions and 11 deletions
|
|
@ -180,7 +180,7 @@ uint32_t llama_hparams::n_layer_kv() const {
|
|||
return res;
|
||||
}
|
||||
|
||||
bool llama_hparams::is_masked_swa(llama_pos p0, llama_pos p1) const {
|
||||
bool llama_hparams::is_masked_swa(uint32_t n_swa, llama_swa_type swa_type, llama_pos p0, llama_pos p1) {
|
||||
assert(p0 >= 0 && p1 >= 0);
|
||||
|
||||
switch (swa_type) {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue