(our, oai, dst2src=False)
| 413 | |
| 414 | |
| 415 | def load_transformer_layer(our, oai, dst2src=False): |
| 416 | load_weights(oai.ln_1, our.input_layernorm, dst2src) |
| 417 | load_weights(oai.ln_2, our.post_attention_layernorm, dst2src) |
| 418 | load_mlp(our.mlp, oai.mlp, dst2src) |
| 419 | load_attention(our.attention, oai.attn, dst2src) |
| 420 | |
| 421 | |
| 422 | def move_weights(our, oai, dst2src=False): |
no test coverage detected