process_assets.py， line 231，ValueError: matmul: Input operand 1 has a mismatch in its core dimension 0, with gufunc signature (n?,k),(k,m?)->(n?,m?) (size 768 is different from 512)

chn-lee-yumi / MaterialSearch

AI语义搜索本地素材。以图搜图、查找本地素材、根据文字描述匹配画面、视频帧搜索、根据画面描述搜索视频。Semantic search. Search local photos and videos through natural language.

GNU General Public License v3.0

863 stars 117 forks source link

process_assets.py， line 231，ValueError: matmul: Input operand 1 has a mismatch in its core dimension 0, with gufunc signature (n?,k),(k,m?)->(n?,m?) (size 768 is different from 512) #43

Closed Mr-z-j closed 11 months ago

Mr-z-j commented 11 months ago

我将环境变量中的模型修改之后报错了

Mr-z-j commented 11 months ago

删除数据库重新扫描后，搜索可以用了。

但是扫描的过程中，8G显卡提示OOM

chn-lee-yumi commented 11 months ago

尝试减少SCAN_PROCESS_BATCH_SIZE看看

Mr-z-j commented 11 months ago

尝试减少SCAN_PROCESS_BATCH_SIZE看看

好的，我先试试。

另外，我在学习源码的过程中发现，这个余弦相似度的计算是直接将数据库里所有的数据全都返回了，然后遍历计算出分数，假如图片数据过多的话，比如说10万张图片，这个速度是不是会越来越慢？？

chn-lee-yumi commented 11 months ago

另外，我在学习源码的过程中发现，这个余弦相似度的计算是直接将数据库里所有的数据全都返回了，然后遍历计算出分数，假如图片数据过多的话，比如说10万张图片，这个速度是不是会越来越慢？？

很明显是的

muxuan1978 commented 11 months ago

如果换成向量数据库能不能提升效率？

chn-lee-yumi commented 11 months ago

如果换成向量数据库能不能提升效率？

应该是可以的

chn-lee-yumi commented 11 months ago

You haven't reply to this issue for 2 weeks. I think your issue have been solved, so I will close this issue. If you have any other questions, please feel free to open a new one.