Skip to content

Commit 8e2f087

Browse files
committed
fix ci
1 parent 268935b commit 8e2f087

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

modules/launch_utils.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -628,6 +628,13 @@ def prepare_environment():
628628
run_pip(f'install -r "{requirements_file}"', "requirements")
629629
startup_timer.record("install requirements")
630630

631+
if not os.path.isfile(requirements_file_for_npu):
632+
requirements_file_for_npu = os.path.join(script_path, requirements_file_for_npu)
633+
634+
if "torch_npu" in torch_command and not requirements_met(requirements_file_for_npu):
635+
run_pip(f"install -r \"{requirements_file_for_npu}\"", "requirements_for_npu")
636+
startup_timer.record("install requirements_for_npu")
637+
631638
if args.skip_ort:
632639
print("Skipping onnxruntime installation.")
633640
else:
@@ -651,16 +658,6 @@ def prepare_environment():
651658
from modules.onnx_impl import initialize_olive
652659
initialize_olive()
653660

654-
from modules import devices
655-
devices.backend = backend
656-
657-
if not os.path.isfile(requirements_file_for_npu):
658-
requirements_file_for_npu = os.path.join(script_path, requirements_file_for_npu)
659-
660-
if "torch_npu" in torch_command and not requirements_met(requirements_file_for_npu):
661-
run_pip(f"install -r \"{requirements_file_for_npu}\"", "requirements_for_npu")
662-
startup_timer.record("install requirements_for_npu")
663-
664661
if not args.skip_install:
665662
run_extensions_installers(settings_file=args.ui_settings_file)
666663

@@ -676,6 +673,9 @@ def prepare_environment():
676673
print("Exiting because of --exit argument")
677674
exit(0)
678675

676+
from modules import devices
677+
devices.backend = backend
678+
679679

680680
def configure_for_tests():
681681
if "--api" not in sys.argv:

0 commit comments

Comments
 (0)