rm require_version_examples (#12088)

2021-06-09 11:02:52 -07:00
parent d1500d9151
commit 61e191987d
6 changed files with 9 additions and 28 deletions
--- a/examples/research_projects/rag/lightning_base.py
+++ b/examples/research_projects/rag/lightning_base.py
@@ -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,