def get_model_list(directory_path): """ :param directory_path: :return: """ import os model_list: list = [] valid_extensions = { '.ckpt', '.pt', '.pth', '.safetensors', '.bin' } for filename in os.listdir(directory_path): if os.path.splitext(filename)[1] in valid_extensions: name_without_extension = os.path.splitext(filename)[0] file_path = os.path.join(directory_path, filename) # model_list.append((name_without_extension, file_path)) model_list.append(file_path) print('\033[34mFILE: ' + file_path + '\033[0m') return model_list