Akjava commited on
Commit
4fec5dd
·
1 Parent(s): 3b9712f
Files changed (1) hide show
  1. upscale_image.py +3 -2
upscale_image.py CHANGED
@@ -17,14 +17,15 @@ def execute(input_image):
17
  pipe.vae.enable_tiling()
18
  # 画像のパスとプロンプト
19
  prompt = "beautiful girl"
20
- first_resize_w = 128
21
- first_resize_h = 128
22
 
23
  # 画像の読み込みとリサイズ
24
  image = input_image#.convert("RGB")
25
  low_res_img = image
26
  if first_resize_w!=0 and first_resize_h!=0:
27
  low_res_img = image.resize((first_resize_w, first_resize_h))
 
28
  upscaled_image = upscale(pipe, prompt, low_res_img)
29
  return upscaled_image
30
 
 
17
  pipe.vae.enable_tiling()
18
  # 画像のパスとプロンプト
19
  prompt = "beautiful girl"
20
+ first_resize_w = 64
21
+ first_resize_h = 64
22
 
23
  # 画像の読み込みとリサイズ
24
  image = input_image#.convert("RGB")
25
  low_res_img = image
26
  if first_resize_w!=0 and first_resize_h!=0:
27
  low_res_img = image.resize((first_resize_w, first_resize_h))
28
+
29
  upscaled_image = upscale(pipe, prompt, low_res_img)
30
  return upscaled_image
31