github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added sumbmodules

Browse source
This commit is contained in:
leno3003 2022-03-30 18:05:26 +02:00
parent 3e50e37e69
commit 9410e20984
2 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
core/interact/interact.py
View file

@ -339,7 +339,6 @@ class InteractBase(object):
def input_bool(self, s, default_value, help_message=None, answer_key=None): def input_bool(self, s, default_value, help_message=None, answer_key=None):
answer_key = answer_key if answer_key is not None else s answer_key = answer_key if answer_key is not None else s
s_base = s
s = f"[{yn_str[default_value]}] {s} ( y/n" s = f"[{yn_str[default_value]}] {s} ( y/n"
if help_message is not None: if help_message is not None:
@ -348,7 +347,7 @@ class InteractBase(object):
while True: while True:
try: try:
inp = self.input(s_base, answer_key) inp = self.input(s, answer_key)
if len(inp) == 0: if len(inp) == 0:
raise ValueError("") raise ValueError("")
@ -363,7 +362,6 @@ class InteractBase(object):
def input_str(self, s, default_value, valid_list=None, show_default_value=True, help_message=None, answer_key=None): def input_str(self, s, default_value, valid_list=None, show_default_value=True, help_message=None, answer_key=None):
answer_key = answer_key if answer_key is not None else s answer_key = answer_key if answer_key is not None else s
s_base = s
if show_default_value and default_value is not None: if show_default_value and default_value is not None:
s = f"[{default_value}] {s}" s = f"[{default_value}] {s}"
@ -386,7 +384,7 @@ class InteractBase(object):
while True: while True:
try: try:
inp = self.input(s_base, answer_key) inp = self.input(s, answer_key)
if len(inp) == 0: if len(inp) == 0:
if default_value is None: if default_value is None:

2
models/ModelBase.py
View file

@ -80,7 +80,7 @@ class ModelBase(object):
s += "- latest" s += "- latest"
io.log_info (s) io.log_info (s)
inp = io.input_str(f"", "0", show_default_value=False, "5" ) inp = io.input_str(f"", "0", show_default_value=False, answer_key = "5")
model_idx = -1 model_idx = -1
try: try:
model_idx = np.clip ( int(inp), 0, len(saved_models_names)-1 ) model_idx = np.clip ( int(inp), 0, len(saved_models_names)-1 )