@@ -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
680680def configure_for_tests ():
681681 if "--api" not in sys .argv :
0 commit comments