From 82a68593bb6c6abbc321c6329a68633aabe97b9d Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Wed, 21 Jun 2023 23:08:34 +0800
Subject: [PATCH] exact avoidOverflow

---
 frontend/src/utils/index.tsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx
index 4707aee..426bc14 100644
--- a/frontend/src/utils/index.tsx
+++ b/frontend/src/utils/index.tsx
@@ -126,7 +126,9 @@ export const getStrategy = (modelConfig: ModelConfig | undefined = undefined) =>
   let params: ModelParameters;
   if (modelConfig) params = modelConfig.modelParameters;
   else params = commonStore.getCurrentModelConfig().modelParameters;
-  const avoidOverflow = params.modelName.toLowerCase().includes('world') && params.precision !== 'fp32';
+  const modelName = params.modelName.toLowerCase();
+  const avoidOverflow = params.precision !== 'fp32' && modelName.includes('world') && (modelName.includes('0.1b') || modelName.includes('0.4b') ||
+    modelName.includes('1.5b') || modelName.includes('1b5'));
   let strategy = '';
   switch (params.device) {
     case 'CPU':