saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-20 19:12:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove use_cdn when loading pretrained model (#5900)

Browse source
This commit is contained in:
Ye Wang 2020-11-23 14:26:55 -08:00 committed by GitHub
parent 208f4c1d3c
commit 3d5b48a894
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/python/tools/transformers/onnx_exporter.py
View file

@ -247,9 +247,7 @@ def load_pretrained_model(model_name, config, cache_dir, custom_model_class, is_
transformers_module = __import__("transformers", fromlist=[model_class_name])
model_class = getattr(transformers_module, model_class_name)
use_cdn = False if model_name == 't5-11b' else True
return model_class.from_pretrained(model_name, config=config, cache_dir=cache_dir, use_cdn=use_cdn)
return model_class.from_pretrained(model_name, config=config, cache_dir=cache_dir)
def load_pt_model(model_name, model_class, cache_dir):