diff --git a/aiserver.py b/aiserver.py index e79c0272..91600cb9 100644 --- a/aiserver.py +++ b/aiserver.py @@ -5077,6 +5077,8 @@ def tpu_raw_generate( sampler_order=koboldai_vars.sampler_order, ) + genout = np.array(genout) + return genout def torch_raw_generate(