No retrieve_caps function in retrieve_caps.py

[KyriaAnnwyn]

Error in SmallCap_demo.ipynb

NameError Traceback (most recent call last) Cell In [6], line 10 7 with torch.no_grad(): 8 image_embedding = retrieval_model.encode_image(pixel_values_retrieval.unsqueeze(0)).cpu().numpy() ---> 10 nns = retrieve_caps(image_embedding, retrieval_index)[0] 11 caps = [captions[i] for i in nns][:4] 13 # prepare prompt

NameError: name 'retrieve_caps' is not defined

[0215Tyj]

I have the same problem, did you solve it?

[YovaKem]

It's been fixed.