From 3708afaf217f322bc37b9c1c906f72c15721452b Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Mon, 25 Nov 2024 08:20:49 -0800
Subject: [PATCH] [Electron] Add fp32 and fp64 to unet inference precision
 options (#1678)

---
 src/constants/serverConfig.ts | 2 ++
 src/types/serverArgs.ts       | 1 +
 2 files changed, 3 insertions(+)

diff --git a/src/constants/serverConfig.ts b/src/constants/serverConfig.ts
index 0eb8a273a2..75c70ee19d 100644
--- a/src/constants/serverConfig.ts
+++ b/src/constants/serverConfig.ts
@@ -163,6 +163,8 @@ export const SERVER_CONFIG_ITEMS: ServerConfig<any>[] = [
     type: 'combo',
     options: [
       FloatingPointPrecision.AUTO,
+      FloatingPointPrecision.FP64,
+      FloatingPointPrecision.FP32,
       FloatingPointPrecision.FP16,
       FloatingPointPrecision.BF16,
       FloatingPointPrecision.FP8E4M3FN,
diff --git a/src/types/serverArgs.ts b/src/types/serverArgs.ts
index aeb43ea49e..011f030d34 100644
--- a/src/types/serverArgs.ts
+++ b/src/types/serverArgs.ts
@@ -40,6 +40,7 @@ export enum CudaMalloc {
 
 export enum FloatingPointPrecision {
   AUTO = 'auto',
+  FP64 = 'fp64',
   FP32 = 'fp32',
   FP16 = 'fp16',
   BF16 = 'bf16',