"...composable_kernel_onnx.git" did not exist on "ccc4a1d365999a3e15623f490314e66c2d671389"
Unverified Commit cd360dcb authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[examples] minimal version requirement run-time check in PL (#8133)


Co-authored-by: default avatarSam Shleifer <sshleifer@gmail.com>
parent 971c638e
...@@ -7,6 +7,7 @@ from typing import Any, Dict ...@@ -7,6 +7,7 @@ from typing import Any, Dict
import pytorch_lightning as pl import pytorch_lightning as pl
from pytorch_lightning.utilities import rank_zero_info from pytorch_lightning.utilities import rank_zero_info
import pkg_resources
from transformers import ( from transformers import (
AdamW, AdamW,
AutoConfig, AutoConfig,
...@@ -32,6 +33,15 @@ from transformers.optimization import ( ...@@ -32,6 +33,15 @@ from transformers.optimization import (
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
try:
pkg = "pytorch_lightning"
min_ver = "1.0.4"
pkg_resources.require(f"{pkg}>={min_ver}")
except pkg_resources.VersionConflict:
logger.warning(
f"{pkg}>={min_ver} is required for a normal functioning of this module, but found {pkg}=={pkg_resources.get_distribution(pkg).version}. Try pip install -r examples/requirements.txt"
)
MODEL_MODES = { MODEL_MODES = {
"base": AutoModel, "base": AutoModel,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment