Przeglądaj źródła

llama : fix restoring the number of outputs from state files (#6687)

compilade 1 rok temu
rodzic
commit
132f55795e
1 zmienionych plików z 2 dodań i 0 usunięć
  1. 2 0
      llama.cpp

+ 2 - 0
llama.cpp

@@ -15478,6 +15478,8 @@ size_t llama_state_set_data(struct llama_context * ctx, const uint8_t * src) {
                 GGML_ASSERT((uint32_t) id < ctx->cparams.n_batch);
                 ctx->output_ids[id] = i;
             }
+
+            ctx->n_outputs = n_outputs;
         }
     }