(raw_sd, key_map)
| 220 | } |
| 221 | |
| 222 | def sd_map_replace(raw_sd, key_map): |
| 223 | sd = {} |
| 224 | for k,v in raw_sd.items(): |
| 225 | for s,d in key_map.items(): |
| 226 | k = k.replace(s,d) |
| 227 | sd[k] = v |
| 228 | return sd |
| 229 | |
| 230 | def llama_permute(raw_sd, n_head, n_head_kv): |
| 231 | # Reverse version of LlamaModel.permute in llama.cpp convert script |
no outgoing calls
no test coverage detected