Skip to content

Commit e354525

Browse files
author
Guang Yang
committed
Add option to skip overriding torch deps
1 parent e1d9cd1 commit e354525

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

install_dev.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import argparse
12
import subprocess
23
import sys
34

@@ -46,11 +47,20 @@ def install_dep_from_source():
4647

4748
def main():
4849
"""Install optimum-executorch in dev mode with nightly dependencies"""
50+
parser = argparse.ArgumentParser()
51+
parser.add_argument(
52+
"--skip_override_torch",
53+
action="store_true",
54+
help="Skip installation of nightly executorch and torch dependencies",
55+
)
56+
args = parser.parse_args()
57+
4958
# Install package with dev extras
5059
subprocess.check_call([sys.executable, "-m", "pip", "install", ".[dev]"])
5160

5261
# Install nightly dependencies
53-
install_torch_nightly_deps()
62+
if not args.skip_override_torch:
63+
install_torch_nightly_deps()
5464

5565
# Install source dependencies
5666
install_dep_from_source()

0 commit comments

Comments
 (0)