wsj1995 commited on
Commit
187b46f
·
1 Parent(s): b214db1

feat: 修改运行

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. facefusion/installer.py +1 -1
app.py CHANGED
@@ -2,9 +2,9 @@
2
 
3
  import os
4
 
5
- os.environ['OMP_NUM_THREADS'] = '1'
6
 
7
- from facefusion import core
8
 
9
  if __name__ == '__main__':
10
- core.cli()
 
2
 
3
  import os
4
 
5
+ os.environ['SYSTEM_VERSION_COMPAT'] = '0'
6
 
7
+ from facefusion import installer
8
 
9
  if __name__ == '__main__':
10
+ installer.cli()
facefusion/installer.py CHANGED
@@ -27,7 +27,7 @@ if is_windows():
27
  def cli() -> None:
28
  signal.signal(signal.SIGINT, lambda signal_number, frame: sys.exit(0))
29
  program = ArgumentParser(formatter_class = lambda prog: HelpFormatter(prog, max_help_position = 50))
30
- program.add_argument('--onnxruntime', help = wording.get('help.install_dependency').format(dependency = 'onnxruntime'), choices = ONNXRUNTIMES.keys(), required = True)
31
  program.add_argument('--skip-conda', help = wording.get('help.skip_conda'), action = 'store_true')
32
  program.add_argument('-v', '--version', version = metadata.get('name') + ' ' + metadata.get('version'), action = 'version')
33
  run(program)
 
27
  def cli() -> None:
28
  signal.signal(signal.SIGINT, lambda signal_number, frame: sys.exit(0))
29
  program = ArgumentParser(formatter_class = lambda prog: HelpFormatter(prog, max_help_position = 50))
30
+ program.add_argument('--onnxruntime',default='cuda', help = wording.get('help.install_dependency').format(dependency = 'onnxruntime'), choices = ONNXRUNTIMES.keys(), required = True)
31
  program.add_argument('--skip-conda', help = wording.get('help.skip_conda'), action = 'store_true')
32
  program.add_argument('-v', '--version', version = metadata.get('name') + ' ' + metadata.get('version'), action = 'version')
33
  run(program)