chore: make steps enums more strict

This commit is contained in:
Manuel Schmid 2024-02-18 16:14:24 +01:00
parent c668228fe8
commit 36f6715512
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
1 changed files with 3 additions and 3 deletions

View File

@ -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