quick fic of model path in output name

This commit is contained in:
DoctorPopi 2023-11-07 20:50:18 +01:00
parent 0a09610c2c
commit 8af8e6b6ca

View File

@ -1141,7 +1141,7 @@ def generate_tortoise(**kwargs):
def get_name(line=0, candidate=0, combined=False, seed=None, model_path=None):
filename_with_extension = os.path.basename(model_path)
model_name, _ = os.path.splitext(filename_with_extension)
model_part = f"model-{model_name}" if model is not None else "No_model_found"
model_part = f"model-{model_name}" if model_path is not None else "No_model_found"
seed_part = f"_seed-{seed}" if seed is not None else ""
name = f"{idx}_{seed_part}_{model_part}"
if combined: