SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

rm require_version_examples (#12088)

Browse Source
This commit is contained in:
Stas Bekman
2021-06-09 11:02:52 -07:00
committed by GitHub
parent d1500d9151
commit 61e191987d
6 changed files with 9 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/legacy/pytorch-lightning/lightning_base.py
View File

@@ -28,12 +28,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"base": AutoModel,

4
examples/research_projects/rag-end2end-retriever/lightning_base.py
View File

@@ -29,12 +29,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"base": AutoModel,

4
examples/research_projects/rag/lightning_base.py
View File

@@ -28,12 +28,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"base": AutoModel,

4
examples/research_projects/seq2seq-distillation/lightning_base.py
View File

@@ -28,12 +28,12 @@ from transformers.optimization import (
get_linear_schedule_with_warmup,
get_polynomial_decay_schedule_with_warmup,
)
from transformers.utils.versions import require_version_examples
from transformers.utils.versions import require_version
logger = logging.getLogger(__name__)
require_version_examples("pytorch_lightning>=1.0.4")
require_version("pytorch_lightning>=1.0.4")
MODEL_MODES = {
"base": AutoModel,