Gameur/Beta
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Voice et bot modif

Browse source
This commit is contained in:
pi 2026-06-16 17:09:34 +00:00
parent 189d56026b
commit 7333a22bcd
10774 changed files with 634644 additions and 933308 deletions
Download patch file Download diff file Expand all files Collapse all files

16
venv/lib/python3.12/site-packages/onnxruntime/transformers/benchmark_helper.py
View file

@ -18,7 +18,6 @@ from enum import Enum
from time import sleep
from typing import Any
import coloredlogs
import numpy
import torch
import transformers
@ -112,12 +111,9 @@ def create_onnxruntime_session(
elif use_gpu:
if provider == "dml":
providers = ["DmlExecutionProvider", "CPUExecutionProvider"]
elif provider == "rocm":
providers = ["ROCMExecutionProvider", "CPUExecutionProvider"]
elif provider == "migraphx":
providers = [
"MIGraphXExecutionProvider",
"ROCMExecutionProvider",
"CPUExecutionProvider",
]
elif provider == "cuda" or provider is None:
@ -150,12 +146,12 @@ def create_onnxruntime_session(
def setup_logger(verbose=True):
if verbose:
coloredlogs.install(
level="DEBUG",
fmt="[%(filename)s:%(lineno)s - %(funcName)20s()] %(message)s",
logging.basicConfig(
format="[%(filename)s:%(lineno)s - %(funcName)20s()] %(message)s",
level=logging.DEBUG,
)
else:
coloredlogs.install(fmt="%(message)s")
logging.basicConfig(format="%(message)s", level=logging.INFO)
logging.getLogger("transformers").setLevel(logging.WARNING)
@ -174,8 +170,8 @@ def prepare_environment(cache_dir, output_dir, use_gpu, provider=None):
else:
assert not set(onnxruntime.get_available_providers()).isdisjoint(
["CUDAExecutionProvider", "ROCMExecutionProvider", "MIGraphXExecutionProvider"]
), "Please install onnxruntime-gpu package, or install ROCm support, to test GPU inference."
["CUDAExecutionProvider", "MIGraphXExecutionProvider"]
), "Please install onnxruntime-gpu package, or install migraphx, to test GPU inference."
logger.info(f"PyTorch Version:{torch.__version__}")
logger.info(f"Transformers Version:{transformers.__version__}")