エントリーの編集
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
GPT-2を使おうとしてエラーが出た話 - Qiita
記事へのコメント0件
- 注目コメント
- 新着コメント
このエントリーにコメントしてみましょう。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています
- バナー広告なし
- ミュート機能あり
- ダークモード搭載
関連記事
GPT-2を使おうとしてエラーが出た話 - Qiita
やりたかったこと PythonでFlaskを使い、リクエストを送るとGPT-2の推論結果を返してくるWebAPIを作りた... やりたかったこと PythonでFlaskを使い、リクエストを送るとGPT-2の推論結果を返してくるWebAPIを作りたかった 問題が発生したコード from transformers import T5Tokenizer, AutoModelForCausalLM import torch from flask import Flask, request print('モデルを読み込みます。数十秒お待ちください。') model_name = 'rinna/japanese-gpt2-medium' # model_name = 'rinna/japanese-gpt-1b' # model_name = 'rinna/japanese-gpt2-medium' # model_name = 'rinna/japanese-gpt-neox-small' tokenizer = T5To