@@ -903,6 +903,7 @@ class llama_sampler_seq_config(ctypes.Structure):
903903# uint32_t n_ubatch; // physical maximum batch size
904904# uint32_t n_seq_max; // max number of sequences (i.e. distinct states for recurrent models)
905905# uint32_t n_rs_seq; // number of recurrent-state snapshots per seq for rollback (0 = no rollback) [EXPERIMENTAL]
906+ # uint32_t n_outputs_max; // max outputs in a ubatch (0 = n_batch)
906907# int32_t n_threads; // number of threads to use for generation
907908# int32_t n_threads_batch; // number of threads to use for batch processing
908909
@@ -958,6 +959,7 @@ class llama_context_params(ctypes.Structure):
958959 n_ubatch (int): physical maximum batch size
959960 n_seq_max (int): max number of sequences (i.e. distinct states for recurrent models)
960961 n_rs_seq (int): number of recurrent-state snapshots per sequence for rollback
962+ n_outputs_max (int): max outputs in a ubatch, 0 = n_batch
961963 n_threads (int): number of threads to use for generation
962964 n_threads_batch (int): number of threads to use for batch processing
963965 ctx_type (int): context type, from `enum llama_context_type`
@@ -995,6 +997,7 @@ class llama_context_params(ctypes.Structure):
995997 n_ubatch : int
996998 n_seq_max : int
997999 n_rs_seq : int
1000+ n_outputs_max : int
9981001 n_threads : int
9991002 n_threads_batch : int
10001003 ctx_type : int
@@ -1031,6 +1034,7 @@ class llama_context_params(ctypes.Structure):
10311034 ("n_ubatch" , ctypes .c_uint32 ),
10321035 ("n_seq_max" , ctypes .c_uint32 ),
10331036 ("n_rs_seq" , ctypes .c_uint32 ),
1037+ ("n_outputs_max" , ctypes .c_uint32 ),
10341038 ("n_threads" , ctypes .c_int32 ),
10351039 ("n_threads_batch" , ctypes .c_int32 ),
10361040 ("ctx_type" , ctypes .c_int ),
0 commit comments