catpokemon commited on
Commit
541f05f
·
verified ·
1 Parent(s): ea01f09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -32,12 +32,12 @@ pipe = FluxKontextPipeline.from_pretrained(
32
  torch_dtype=torch.bfloat16,
33
  ).to("cuda")
34
 
35
- # 변경점: 숫자를 가장 가까운 8의 배수로 만들어주는 함수 추가
36
- def to_multiple_of_8(n):
37
  """
38
- Rounds a number to the nearest multiple of 8.
39
  """
40
- return int(np.round(n / 8.0) * 8.0)
41
 
42
  @spaces.GPU
43
  def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
@@ -86,10 +86,10 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
86
  if input_image:
87
  input_image = input_image.convert("RGB")
88
 
89
- # 변경점: 모델의 요구사항에 맞게 이미지 크기를 8의 배수로 조정합니다.
90
  original_width, original_height = input_image.size
91
- new_width = to_multiple_of_8(original_width)
92
- new_height = to_multiple_of_8(original_height)
93
 
94
  # 조정된 크기로 이미지를 리사이즈합니다.
95
  resized_image = input_image.resize((new_width, new_height))
@@ -103,7 +103,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
103
  ).images[0]
104
 
105
  else:
106
- # 변경점: 텍스트-이미지 생성 시에는 크기를 명시해주는 것이 좋습니다.
107
  image = pipe(
108
  prompt=prompt,
109
  width=1024,
 
32
  torch_dtype=torch.bfloat16,
33
  ).to("cuda")
34
 
35
+ # 변경점: 숫자를 가장 가까운 16의 배수로 만들어주는 함수로 수정
36
+ def to_multiple_of_16(n):
37
  """
38
+ Rounds a number to the nearest multiple of 16.
39
  """
40
+ return int(np.round(n / 16.0) * 16.0)
41
 
42
  @spaces.GPU
43
  def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
 
86
  if input_image:
87
  input_image = input_image.convert("RGB")
88
 
89
+ # 변경점: 모델의 요구사항에 맞게 이미지 크기를 16의 배수로 조정합니다.
90
  original_width, original_height = input_image.size
91
+ new_width = to_multiple_of_16(original_width)
92
+ new_height = to_multiple_of_16(original_height)
93
 
94
  # 조정된 크기로 이미지를 리사이즈합니다.
95
  resized_image = input_image.resize((new_width, new_height))
 
103
  ).images[0]
104
 
105
  else:
106
+ # 텍스트-이미지 생성 시에는 크기를 명시해주는 것이 좋습니다.
107
  image = pipe(
108
  prompt=prompt,
109
  width=1024,