Spaces:
Sleeping
Sleeping
Kimilhee
commited on
Commit
·
4e5f4b8
1
Parent(s):
1ad3919
custom 크기 이미지 생성시 에러 수정.
Browse files- image_generator.py +4 -0
image_generator.py
CHANGED
@@ -10,6 +10,7 @@ from config import (
|
|
10 |
LOW_PRICE_OPTION,
|
11 |
HIGH_PRICE,
|
12 |
LOW_PRICE,
|
|
|
13 |
)
|
14 |
|
15 |
|
@@ -178,6 +179,9 @@ def genPromptAndImage(
|
|
178 |
lowImageCnt,
|
179 |
)
|
180 |
|
|
|
|
|
|
|
181 |
if fluxPrompt == "":
|
182 |
fluxPrompt = genFluxPrompt(userPrompt, additionalComment, ratio, imageStyle)
|
183 |
|
|
|
10 |
LOW_PRICE_OPTION,
|
11 |
HIGH_PRICE,
|
12 |
LOW_PRICE,
|
13 |
+
CUSTOM,
|
14 |
)
|
15 |
|
16 |
|
|
|
179 |
lowImageCnt,
|
180 |
)
|
181 |
|
182 |
+
if ratio == CUSTOM:
|
183 |
+
ratio = "custom"
|
184 |
+
|
185 |
if fluxPrompt == "":
|
186 |
fluxPrompt = genFluxPrompt(userPrompt, additionalComment, ratio, imageStyle)
|
187 |
|