| 959 | ds4_tokens_free(&prefix); |
| 960 | free(live_logits); |
| 961 | free(fresh_logits); |
| 962 | return 1; |
| 963 | } |
| 964 | |
| 965 | cli_prefill_progress progress = { |
| 966 | .base_tokens = 0, |
| 967 | .input_tokens = prompt->len, |
| 968 | .use_color = ds4_log_is_tty(stderr), |
| 969 | }; |
| 970 | ds4_session_set_progress(live, cli_prefill_progress_cb, &progress); |
| 971 | ds4_session_set_display_progress(live, |
| 972 | progress.use_color ? cli_prefill_progress_cb : NULL, |
| 973 | progress.use_color ? &progress : NULL); |
| 974 | if (ds4_session_sync(live, prompt, err, sizeof(err)) != 0) { |
| 975 | ds4_session_set_progress(live, NULL, NULL); |