Kimilhee commited on
Commit
4e5f4b8
·
1 Parent(s): 1ad3919

custom 크기 이미지 생성시 에러 수정.

Browse files
Files changed (1) hide show
  1. 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