Skip to content

Commit de87a6e

Browse files
committed
deprecate transformers model examples
1 parent 727db0d commit de87a6e

File tree

9 files changed

+44
-0
lines changed

9 files changed

+44
-0
lines changed

onnxruntime/python/tools/transformers/models/gpt2/convert_to_onnx.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import os
2222
import shutil
2323
import sys
24+
import warnings
2425
from pathlib import Path
2526

2627
import numpy
@@ -243,6 +244,13 @@ def get_latency_name(batch_size, sequence_length, past_sequence_length):
243244

244245

245246
def main(argv=None, experiment_name: str = "", run_id: str = "0", csv_filename: str = "gpt2_parity_results.csv"):
247+
warnings.warn(
248+
"This example is deprecated. Use the Olive recipe instead: "
249+
"https://github.com/microsoft/olive-recipes/tree/main",
250+
DeprecationWarning,
251+
stacklevel=2,
252+
)
253+
246254
result = {}
247255
if version.parse(transformers_version) < version.parse(
248256
"3.1.0"

onnxruntime/python/tools/transformers/models/llama/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
> **Deprecated:** This example is deprecated. Use the Olive recipes instead: https://github.com/microsoft/olive-recipes/tree/main
2+
13
# Contents
24
- [LLaMA-2](#llama-2)
35
- [Prerequisites](#prerequisites)

onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import subprocess
1313
import sys
1414
import tempfile
15+
import warnings
1516
from itertools import chain
1617

1718
import onnx
@@ -801,6 +802,12 @@ def get_args():
801802

802803

803804
def main():
805+
warnings.warn(
806+
"This example is deprecated. Use the Olive recipe instead: "
807+
"https://github.com/microsoft/olive-recipes/tree/main",
808+
DeprecationWarning,
809+
stacklevel=2,
810+
)
804811
if version.parse(torch.__version__) < version.parse("2.2.0"):
805812
logger.error(f"Detected PyTorch version {torch.__version__}. Please upgrade and use v2.2.0 or newer.")
806813
return

onnxruntime/python/tools/transformers/models/phi2/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
> **Deprecated:** This example is deprecated. Use the Olive recipes instead: https://github.com/microsoft/olive-recipes/tree/main
2+
13
# Phi2 Optimizations
24
## Prerequisites
35
A Linux machine for [TorchDynamo-based ONNX Exporter](https://pytorch.org/docs/stable/onnx.html#torchdynamo-based-onnx-exporter)\

onnxruntime/python/tools/transformers/models/phi2/convert_to_onnx.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import argparse
88
import logging
99
import os
10+
import warnings
1011
from pathlib import Path
1112

1213
import onnx
@@ -375,6 +376,12 @@ def parse_arguments():
375376

376377

377378
def main():
379+
warnings.warn(
380+
"This example is deprecated. Use the Olive recipe instead: "
381+
"https://github.com/microsoft/olive-recipes/tree/main",
382+
DeprecationWarning,
383+
stacklevel=2,
384+
)
378385
args = parse_arguments()
379386

380387
device = torch.device("cuda", args.device_id) if torch.cuda.is_available() else torch.device("cpu")

onnxruntime/python/tools/transformers/models/stable_diffusion/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
> **Deprecated:** This example is deprecated. Use the Olive recipes instead: https://github.com/microsoft/olive-recipes/tree/main
2+
13
# Stable Diffusion GPU Optimization
24

35
ONNX Runtime uses the following optimizations to speed up Stable Diffusion in CUDA:

onnxruntime/python/tools/transformers/models/stable_diffusion/optimize_pipeline.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import os
2121
import shutil
2222
import tempfile
23+
import warnings
2324
from pathlib import Path
2425

2526
import coloredlogs
@@ -569,6 +570,12 @@ def parse_arguments(argv: list[str] | None = None):
569570

570571

571572
def main(argv: list[str] | None = None):
573+
warnings.warn(
574+
"This example is deprecated. Use the Olive recipe instead: "
575+
"https://github.com/microsoft/olive-recipes/tree/main",
576+
DeprecationWarning,
577+
stacklevel=2,
578+
)
572579
args = parse_arguments(argv)
573580

574581
logger.info("Arguments: %s", str(args))

onnxruntime/python/tools/transformers/models/whisper/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
> **Deprecated:** This example is deprecated. Use the Olive recipes instead: https://github.com/microsoft/olive-recipes/tree/main
2+
13
# Whisper
24

35
## Prerequisites

onnxruntime/python/tools/transformers/models/whisper/convert_to_onnx.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import argparse
88
import logging
99
import os
10+
import warnings
1011

1112
import onnx
1213
import torch
@@ -493,6 +494,12 @@ def export_onnx_models(
493494

494495

495496
def main(argv=None):
497+
warnings.warn(
498+
"This example is deprecated. Use the Olive recipe instead: "
499+
"https://github.com/microsoft/olive-recipes/tree/main",
500+
DeprecationWarning,
501+
stacklevel=2,
502+
)
496503
args = parse_arguments(argv)
497504

498505
setup_logger(args.verbose)

0 commit comments

Comments
 (0)