Add n_keep parameter to LLama constructor to enable Streaming-LLM

オープン 初心者向け
#954 コメント 1 件 リアクション 4 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

評価

難易度
2/5
見積もり時間
1〜3時間
初心者へのやさしさ
65/100
issue の種類
機能追加
明瞭さ
明確に書かれている
活発さ
停滞
技術スタック
c, python
領域
ai, ai-infra-agents, backend-api-design

調査の方向性

高レベルAPIのLlamaクラスのコンストラクターを確認し、パラメーターがどこで定義されているかを調べます。低レベルAPIの例では、lparamsを指定してllama_load_model_from_fileを呼び出す方法が示されています。コンストラクターにn_keepを追加し、モデルの読み込みパラメーターに渡して、ドキュメント化されていることを確認します。keepパラメーターを指定してモデルを作成し、正しく渡されることを確認してテストします。

索引モデルが issue の本文から書いたものです。

説明

enhancement

A recent paper by Meta/MIT/CMU proposed StreamingLLM, a simple yet efficient solution to enable "infinite" context. Better yet, the implementation in llama.cpp is as trivial as changing the n_keep value with option --keep as discussed in this issue. Unfortunately, the high-level API of llama-cpp-python does not support the keep/n_keep parameter.

It should be simple to add the parameter to the high-level API, ideally in the constructor for class Llama and to pass it along to function llama_cpp.llama_load_model_from_file as part of parameter lparams here.

主要言語
Python
スター
10.6k
フォーク
1.5k
平均マージ
6時間 43分
マージ済み PR（30日）
2

コントリビューションガイド

コントリビューションガイドを開く

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

abetlen/llama-cpp-python のほかの issue

abetlen/llama-cpp-python の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。