diff --git a/aiserver.py b/aiserver.py index 2eb06c32..9ef798cd 100644 --- a/aiserver.py +++ b/aiserver.py @@ -2572,7 +2572,7 @@ def spRequest(filename): -1, tpu_mtj_backend.params["d_model"], ) - vars.sp = tensor + vars.sp = np.float32(tensor) else: vars.sp = torch.from_numpy(tensor)