saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-26 22:35:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix import in transformer optimizer python script (#20091)

Browse source 
### Description
Fix import.

### Motivation and Context
Fix error.
This commit is contained in:
Xiaoyu 2024-03-26 20:16:09 -07:00 committed by GitHub
parent 473434c73f
commit 512c803550
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/transformers/optimizer.py
View file

@ -40,9 +40,9 @@ from onnx_model_t5 import T5OnnxModel
from onnx_model_tnlr import TnlrOnnxModel
from onnx_model_unet import UnetOnnxModel
from onnx_model_vae import VaeOnnxModel
from onnx_utils import extract_raw_data_from_model, has_external_data
import onnxruntime
from onnxruntime.transformers.onnx_utils import extract_raw_data_from_model, has_external_data
logger = logging.getLogger(__name__)