feat: use class PerformanceLoRA instead of strings in config

This commit is contained in:
Manuel Schmid
2024-05-30 00:13:14 +02:00
parent 4127fa410b
commit 9564341fbd
2 changed files with 15 additions and 6 deletions
+8
View File
@@ -115,6 +115,14 @@ class OutputFormat(Enum):
return list(map(lambda c: c.value, cls))
class PerformanceLoRA(Enum):
QUALITY = None
SPEED = None
EXTREME_SPEED = 'sdxl_lcm_lora.safetensors'
LIGHTNING = 'sdxl_lightning_4step_lora.safetensors'
HYPER_SD = 'sdxl_hyper_sd_4step_lora.safetensors'
class Steps(IntEnum):
QUALITY = 60
SPEED = 30