From 77a9226720a6bd8b578a12920a2bce32116f22d8 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 11 Dec 2024 11:47:18 +0530
Subject: [PATCH] Annotate with types for pydantic

---
 ui/easydiffusion/types.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ui/easydiffusion/types.py b/ui/easydiffusion/types.py
index f16e9aa3..2da62537 100644
--- a/ui/easydiffusion/types.py
+++ b/ui/easydiffusion/types.py
@@ -20,8 +20,8 @@ class GenerateImageRequest(BaseModel):
     control_image: Any = None
     control_alpha: Union[float, List[float]] = None
     prompt_strength: float = 0.8
-    preserve_init_image_color_profile = False
-    strict_mask_border = False
+    preserve_init_image_color_profile: bool = False
+    strict_mask_border: bool = False
 
     sampler_name: str = None  # "ddim", "plms", "heun", "euler", "euler_a", "dpm2", "dpm2_a", "lms"
     hypernetwork_strength: float = 0
@@ -100,7 +100,7 @@ class MergeRequest(BaseModel):
     model1: str = None
     ratio: float = None
     out_path: str = "mix"
-    use_fp16 = True
+    use_fp16: bool = True
 
 
 class Image: