From 36f671551274bec4da16c2d235f184bc63b1808f Mon Sep 17 00:00:00 2001
From: Manuel Schmid <manuel.schmid@odt.net>
Date: Sun, 18 Feb 2024 16:14:24 +0100
Subject: [PATCH] chore: make steps enums more strict

---
 modules/flags.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/modules/flags.py b/modules/flags.py
index 278f7000..a32056ba 100644
--- a/modules/flags.py
+++ b/modules/flags.py
@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import IntEnum, Enum
 
 disabled = 'Disabled'
 enabled = 'Enabled'
@@ -93,13 +93,13 @@ lora_count_with_lcm = lora_count + 1
 controlnet_image_count = 4
 
 
-class Steps(Enum):
+class Steps(IntEnum):
     QUALITY = 60
     SPEED = 30
     EXTREME_SPEED = 8
 
 
-class StepsUOV(Enum):
+class StepsUOV(IntEnum):
     QUALITY = 36
     SPEED = 18
     EXTREME_SPEED = 8