coralLight commited on
Commit
c6f1748
·
1 Parent(s): 6cb7a6d
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -918,7 +918,10 @@ def generate_image_with_steps(prompt, negative_prompt, seed, width, height, guid
918
  x_sample = 255. * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
919
  image = Image.fromarray(x_sample.astype(np.uint8))
920
  # base_count += 1
921
-
 
 
 
922
  # image = pipe(
923
  # prompt=prompt,
924
  # negative_prompt=negative_prompt,
 
918
  x_sample = 255. * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
919
  image = Image.fromarray(x_sample.astype(np.uint8))
920
  # base_count += 1
921
+ width, height = image.size
922
+ width = width / 2
923
+ height = height / 2
924
+ image = image.resize((int(width), int(height)))
925
  # image = pipe(
926
  # prompt=prompt,
927
  # negative_prompt=negative_prompt,