タグ

ブックマーク / zenn.dev/octu0 (1)

  • RWKVとsqlite-vssで高速なベクトル検索を作ってみる

    はじめに 最近 langchain を使うようになってきて、OpenAIAPIをちょこちょこ叩くのですが、いかんせん遅い いや十分に早いのだけど、ドキュメントの量があると若干気になってくる速度です そこで、 ローカルLLMとしてrinna を使ってみたりしたのですが、まだまだ遅いです すでに先行して実装例を作ってくれていた RWKVでembedding vectorを計算 の記事と SQLiteでベクトル検索ができる拡張sqlite-vssを試す の記事を大いに参考にしながら RWKV と sqlite-vss を使って高速なベクトル検索を作ってみます RWKVのモデル 今回は学習済みのRavenモデル RWKV-4-Raven-3B-v12-Eng98%-Other2% を使っていきます OpenAItext-embedding-ada-002 モデル では 1536次元と扱いや

    RWKVとsqlite-vssで高速なベクトル検索を作ってみる
  • 1