gpt j api

gpt-j-api ?

GPT-J 言語モデルおよびバリアントと対話するための API!モデルは 2 つの異なる方法で使用およびテストできます。

• Streamlit Web アプリ (http://api.vicgalle.net:8000/)
• 適切な API (http://api.vicgalle.net:5000/docs に記載)

オープン API エンドポイント ?

これらはパブリック API のエンドポイントであり、認証は必要ありません。それぞれをクリックするとパラメータが表示されます。

GPT-Jテキスト生成?

• 生成 : POST /generate/

ゼロショットテキスト分類 (多言語) ?

• 分類 : POST /classify/

APIの使用

• パイソン:

 import requests
context = "In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English."
payload = {
    "context" : context ,
    "token_max_length" : 512 ,
    "temperature" : 1.0 ,
    "top_p" : 0.9 ,
}
response = requests . post ( "http://api.vicgalle.net:5000/generate" , params = payload ). json ()
print ( response )

• Python (ゼロショット分類):

 import requests
payload = { 
    "sequence" : "The movie started slow, but in the end was absolutely amazing!" , 
    "labels" : "positive,neutral,negative" }
response = requests . post ( "http://api.vicgalle.net:5000/classify" , params = payload ). json ()
print ( response )

• バッシュ:

curl -X ' POST ' 
  ' http://api.vicgalle.net:5000/generate?context=In%20a%20shocking%20finding%2C%20scientists%20discovered%20a%20herd%20of%20unicorns%20living%20in%20a%20remote%2C%20previously%20unexplored%20valley%2C%20in%20the%20Andes%20Mountains.%20Even%20more%20surprising%20to%20the%20researchers%20was%20the%20fact%20that%20the%20unicorns%20spoke%20perfect%20English.&token_max_length=512&temperature=1&top_p=0.9 ' 
  -H ' accept: application/json ' 
  -d ' ' 

APIサーバーの導入

TPU VM に SSH 接続するだけです。このコードは、v2-8 と v3-8 の両方のバリアントでテストされました。

まず、要件をインストールし、重みを取得します。

 python3 -m pip install -r requirements.txt
wget https://the-eye.eu/public/AI/GPT-J-6B/step_383500_slim.tar.zstd
sudo apt install zstd
tar -I zstd -xf step_383500_slim.tar.zstd
rm step_383500_slim.tar.zstd

そしてただ走ってください

 python3 serve.py

その後、http://localhost:5000/docs にアクセスして API を使用できます。

streamlit ダッシュボードを導入する

ただ走ってください

 python3 -m streamlit run streamlit_app.py --server.port 8000

接触

ご要望がございましたら、 vgallegoalcala at gmail dot comまで喜んでお手伝いさせていただきます

スポンサー？

このプロジェクトを後援してくださった以下の方々に心より感謝いたします。 <3

• アスペ96

謝辞

TPU Research Cloud (https://sites.research.google/trc/) のサポートに感謝します。