Mizuiro-sakura commited on
Commit
981f30e
1 Parent(s): b5191a8

Update deberta_squad.py

Browse files
Files changed (1) hide show
  1. deberta_squad.py +1 -1
deberta_squad.py CHANGED
@@ -2,7 +2,7 @@ import torch
2
  from transformers import AutoTokenizer
3
 
4
  tokenizer = AutoTokenizer.from_pretrained('ku-nlp/deberta-v2-base-japanese')
5
- model=torch.load('C:\\[modelのあるディレクトリ]\\My_deberta_model_squad.pth') # 学習済みモデルの読み込み
6
 
7
  text={
8
  'context':'私の名前はEIMIです。好きな食べ物は苺です。 趣味は皆さんと会話することです。',
 
2
  from transformers import AutoTokenizer
3
 
4
  tokenizer = AutoTokenizer.from_pretrained('ku-nlp/deberta-v2-base-japanese')
5
+ model=torch.load('C:\\[.pth modelのあるディレクトリ]\\My_deberta_model_squad.pth') # 学習済みモデルの読み込み
6
 
7
  text={
8
  'context':'私の名前はEIMIです。好きな食べ物は苺です。 趣味は皆さんと会話することです。',