| 608 | } |
| 609 | |
| 610 | void gpt2_free(GPT2 *model) { |
| 611 | free(model->params_memory); |
| 612 | free(model->grads_memory); |
| 613 | free(model->m_memory); |
| 614 | free(model->v_memory); |
| 615 | free(model->acts_memory); |
| 616 | free(model->grads_acts_memory); |
| 617 | free(model->inputs); |
| 618 | free(model->targets); |
| 619 | } |
| 620 | |
| 621 | #ifndef TESTING |
| 622 | // if we are TESTING (see test_gpt2.c), we'll skip the int main below |