root commited on
Commit
49f2d34
1 Parent(s): 649468d
Files changed (2) hide show
  1. app.py +4 -4
  2. roop/predictor.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import roop.globals
6
  from roop.core import (
7
  start,
8
  decode_execution_providers,
9
- suggest_max_memory,
10
  suggest_execution_threads,
11
  )
12
  from roop.processors.frame.core import get_frame_processors_modules
@@ -45,9 +45,9 @@ def swap_face(source_file, target_file,doFaceEnhancer):
45
  roop.globals.many_faces = False
46
  roop.globals.video_encoder = "libx264"
47
  roop.globals.video_quality = 18
48
- roop.globals.max_memory = suggest_max_memory()
49
- roop.globals.execution_providers = decode_execution_providers(["cuda"])
50
- roop.globals.execution_threads = suggest_execution_threads()
51
 
52
  print(
53
  "start process",
 
6
  from roop.core import (
7
  start,
8
  decode_execution_providers,
9
+ #suggest_max_memory,
10
  suggest_execution_threads,
11
  )
12
  from roop.processors.frame.core import get_frame_processors_modules
 
45
  roop.globals.many_faces = False
46
  roop.globals.video_encoder = "libx264"
47
  roop.globals.video_quality = 18
48
+ #roop.globals.max_memory = suggest_max_memory()
49
+ roop.globals.execution_providers = decode_execution_providers(["cpu"])
50
+ #roop.globals.execution_threads = suggest_execution_threads()
51
 
52
  print(
53
  "start process",
roop/predictor.py CHANGED
@@ -8,7 +8,7 @@ from roop.typing import Frame
8
 
9
  PREDICTOR = None
10
  THREAD_LOCK = threading.Lock()
11
- MAX_PROBABILITY = 0.85
12
 
13
 
14
  def get_predictor() -> Model:
 
8
 
9
  PREDICTOR = None
10
  THREAD_LOCK = threading.Lock()
11
+ MAX_PROBABILITY = 1
12
 
13
 
14
  def get_predictor() -> Model: