diff --git a/webui.py b/webui.py index 76cbfefe295..f83e11cb6cb 100644 --- a/webui.py +++ b/webui.py @@ -362,11 +362,15 @@ def initial_s3_download(s3_folder, local_folder,cache_dir,mode): #only fetch the first model to download. if mode == 'sd': s3_files = {} - _, file_names = next(iter(fnames_dict.items())) - for fname in file_names: - s3_files[fname] = tmp_s3_files.get(fname) - check_space_s3_download(s3,shared.models_s3_bucket, s3_folder,local_folder, fname, tmp_s3_files.get(fname)[1], mode) - register_models(local_folder,mode) + try: + _, file_names = next(iter(fnames_dict.items())) + for fname in file_names: + s3_files[fname] = tmp_s3_files.get(fname) + check_space_s3_download(s3,shared.models_s3_bucket, s3_folder,local_folder, fname, tmp_s3_files.get(fname)[1], mode) + register_models(local_folder,mode) + except Exception as e: + traceback.print_stack() + print(e) print(f'-----s3_files---{s3_files}') # save the lastest one