diff --git a/modules/flags.py b/modules/flags.py index cc191c92..25b0caae 100644 --- a/modules/flags.py +++ b/modules/flags.py @@ -152,6 +152,10 @@ class Performance(Enum): def list(cls) -> list: return list(map(lambda c: c.value, cls)) + @classmethod + def by_steps(cls, steps: int | str): + return cls[Steps(int(steps)).name] + @classmethod def has_restricted_features(cls, x) -> bool: if isinstance(x, Performance):