TypeError: Sam.__call__() got an unexpected keyword argument 'input_image'

[rohit901]

Hello the demo usage code of sam is giving me this error:
TypeError: Sam.call() got an unexpected keyword argument 'input_image'

Changing line 28 to:

ret = sam_model.apply(
    {'params': params},
    image = input_image,
    point_coords = point_coords,
    point_labels = point_labels,
    padding_mask = padding_mask,
    return_image_embedding=True,
    train=False)

and line 44 to:

ret = sam_model.apply(
    {'params': params},
    image=None,
    point_coords=point_coords,
    point_labels=point_labels,
    padding_mask=padding_mask,
    image_embeddings=cached_image_embedding,
    train=False)

fixes the issue, i.e., passing in "image = " argument when calling the sam_model.

[rohit901]

tagging @xingyizhou