From de755463e34ce9c4318c5121d711ac5f32d1902d Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Tue, 3 Oct 2023 14:27:44 +0800
Subject: [PATCH] improve overflow

---
 frontend/src/utils/index.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx
index 63dfe20..6bf2b3d 100644
--- a/frontend/src/utils/index.tsx
+++ b/frontend/src/utils/index.tsx
@@ -183,7 +183,7 @@ export const getStrategy = (modelConfig: ModelConfig | undefined = undefined) =>
     case 'CUDA':
     case 'CUDA-Beta':
       if (avoidOverflow)
-        strategy = 'cuda fp32 *1 -> ';
+        strategy = params.useCustomCuda ? 'cuda fp16 *1 -> ' : 'cuda fp32 *1 -> ';
       strategy += 'cuda ';
       strategy += params.precision === 'fp16' ? 'fp16' : params.precision === 'int8' ? 'fp16i8' : 'fp32';
       if (params.storedLayers < params.maxStoredLayers)