| 693 | } |
| 694 | |
| 695 | void save_train_state_gguf(struct gguf_context * fctx, struct train_state * train) { |
| 696 | gguf_set_val_u32(fctx, LLM_KV_TRAINING_FILE_VERSION, 1); |
| 697 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_ITERATION_COUNT, train->train_its); |
| 698 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_SAMPLE_COUNT, train->train_samples); |
| 699 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_TOKEN_COUNT, train->train_tokens); |
| 700 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_EPOCH_COUNT, train->train_epochs); |
| 701 | |
| 702 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_SHUFFLE_SAMPLES_HASH, (uint64_t) train->shuffle_samples_hash); |
| 703 | gguf_set_val_str(fctx, LLM_KV_TRAINING_SHUFFLE_RNG_STATE, train->shuffle_rng_state_current.c_str()); |
| 704 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_SHUFFLE_SAMPLE_COUNT, (uint64_t) train->shuffle_sample_count); |
| 705 | gguf_set_val_u64(fctx, LLM_KV_TRAINING_SHUFFLE_NEXT_SAMPLE, (uint64_t) train->shuffle_next_sample); |
| 706 | |
| 707 | save_opt_context_gguf(fctx, train->opt); |
| 708 | } |
| 709 | |
| 710 | |
| 711 | struct llama_file { |
no test coverage detected