Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
1ac6a246
Unverified
Commit
1ac6a246
authored
Apr 03, 2020
by
ahotrod
Committed by
GitHub
Apr 03, 2020
Browse files
Update README.md (#3604)
Update AutoModel & AutoTokernizer loading.
parent
e91692f4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
16 deletions
+2
-16
model_cards/ahotrod/xlnet_large_squad2_512/README.md
model_cards/ahotrod/xlnet_large_squad2_512/README.md
+2
-16
No files found.
model_cards/ahotrod/xlnet_large_squad2_512/README.md
View file @
1ac6a246
...
...
@@ -56,22 +56,8 @@ PyTorch: 1.4.0
TensorFlow: 2.1.0
Python: 3.7.6
```
### Inferencing / prediction works with Transformers v2.4.1, the latest version tested
### Utilize this xlnet_large_squad2_512 fine-tuned model with:
```
python
config_class
,
model_class
,
tokenizer_class
=
\
XLNetConfig
,
XLNetforQuestionAnswering
,
XLNetTokenizer
model_name_or_path
=
"ahotrod/xlnet_large_squad2_512"
config
=
config_class
.
from_pretrained
(
model_name_or_path
)
tokenizer
=
tokenizer_class
.
from_pretrained
(
model_name_or_path
,
do_lower_case
=
True
)
model
=
model_class
.
from_pretrained
(
model_name_or_path
,
config
=
config
)
```
### or the AutoModels (AutoConfig, AutoTokenizer & AutoModel) should also work, however I have yet to use them in my apps & confirm:
```
python
from
transformers
import
AutoConfig
,
AutoTokenizer
,
AutoModel
model_name_or_path
=
"ahotrod/xlnet_large_squad2_512"
config
=
AutoConfig
.
from_pretrained
(
model_name_or_path
)
tokenizer
=
AutoTokenizer
.
from_pretrained
(
model_name_or_path
,
do_lower_case
=
True
)
model
=
AutoModel
.
from_pretrained
(
model_name_or_path
,
config
=
config
)
tokenizer
=
AutoTokenizer
.
from_pretrained
(
"ahotrod/xlnet_large_squad2_512"
)
model
=
AutoModelForQuestionAnswering
.
from_pretrained
(
"ahotrod/xlnet_large_squad2_512"
)
```
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment