Skip to content

Commit

Permalink
nit
Browse files Browse the repository at this point in the history
  • Loading branch information
hh-space-invader committed Nov 21, 2024
1 parent 82e2d4b commit 414a4fe
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions fastembed/image/transform/operators.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,8 +218,7 @@ def _get_pil2ndarray(transforms: list[Transform], config: dict[str, Any]):

@staticmethod
def _get_rescale(transforms: list[Transform], config: dict[str, Any]):
# mode = config.get("image_processor_type", "CLIPImageProcessor")
if config.get("do_rescale", True): # or (mode == "JinaCLIPImageProcessor"):
if config.get("do_rescale", True):
rescale_factor = config.get("rescale_factor", 1 / 255)
transforms.append(Rescale(scale=rescale_factor))

Expand Down

0 comments on commit 414a4fe

Please sign in to comment.