diff --git a/backend-python/requirements.txt b/backend-python/requirements.txt new file mode 100644 index 0000000..c241a89 Binary files /dev/null and b/backend-python/requirements.txt differ diff --git a/backend-python/requirements_versions.txt b/backend-python/requirements_versions.txt new file mode 100644 index 0000000..64bf5c5 Binary files /dev/null and b/backend-python/requirements_versions.txt differ diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx index b88cefe..1a54210 100644 --- a/frontend/src/utils/index.tsx +++ b/frontend/src/utils/index.tsx @@ -131,7 +131,7 @@ export const getStrategy = (modelConfig: ModelConfig | undefined = undefined) => else params = commonStore.getCurrentModelConfig().modelParameters; let strategy = ''; strategy += (params.device === 'CPU' ? 'cpu' : 'cuda') + ' '; - strategy += (params.precision === 'fp16' ? 'fp16' : params.precision === 'int8' ? 'fp16i8' : 'fp32'); + strategy += params.device === 'CPU' ? 'fp32' : (params.precision === 'fp16' ? 'fp16' : params.precision === 'int8' ? 'fp16i8' : 'fp32'); if (params.storedLayers < params.maxStoredLayers) strategy += ` *${params.storedLayers}+`; if (params.enableHighPrecisionForLastLayer)