linoyts HF Staff commited on
Commit
af04736
·
verified ·
1 Parent(s): e3d4b88

Update kontext_pipeline.py

Browse files
Files changed (1) hide show
  1. kontext_pipeline.py +2 -0
kontext_pipeline.py CHANGED
@@ -942,12 +942,14 @@ class FluxKontextPipeline(
942
  image_width, image_height = self.image_processor.get_default_height_width(image[0])
943
  else:
944
  image_width, image_height = self.image_processor.get_default_height_width(image)
 
945
  aspect_ratio = image_width / image_height
946
  if _auto_resize:
947
  # Kontext is trained on specific resolutions, using one of them is recommended
948
  _, image_width, image_height = min(
949
  (abs(aspect_ratio - w / h), w, h) for w, h in PREFERRED_KONTEXT_RESOLUTIONS
950
  )
 
951
  image_width = image_width // multiple_of * multiple_of
952
  image_height = image_height // multiple_of * multiple_of
953
  image = self.image_processor.resize(image, image_height, image_width)
 
942
  image_width, image_height = self.image_processor.get_default_height_width(image[0])
943
  else:
944
  image_width, image_height = self.image_processor.get_default_height_width(image)
945
+ print("orig width, height:", image_width, image_height)
946
  aspect_ratio = image_width / image_height
947
  if _auto_resize:
948
  # Kontext is trained on specific resolutions, using one of them is recommended
949
  _, image_width, image_height = min(
950
  (abs(aspect_ratio - w / h), w, h) for w, h in PREFERRED_KONTEXT_RESOLUTIONS
951
  )
952
+ print("new width, height:", image_width, image_height)
953
  image_width = image_width // multiple_of * multiple_of
954
  image_height = image_height // multiple_of * multiple_of
955
  image = self.image_processor.resize(image, image_height, image_width)