DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-01-08 11:57:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

prevent crash that occurs when changing models.yaml on windows systems

Browse Source 
Windows does not support an atomic `os.rename()` operation. This
PR changes it to `os.replace()`, which does the same thing.
This commit is contained in:
Lincoln Stein 2022-11-25 16:37:54 +00:00
parent dba0280790
commit 8f3f64b22e
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/invoke/model_cache.py
View File

@ -298,7 +298,7 @@ class ModelCache(object):
with open(tmpfile, 'w') as outfile: with open(tmpfile, 'w') as outfile:
outfile.write(self.preamble()) outfile.write(self.preamble())
outfile.write(yaml_str) outfile.write(yaml_str)
os.rename(tmpfile,config_file_path) os.replace(tmpfile,config_file_path)
def preamble(self): def preamble(self):
''' '''

2
ldm/modules/image_degradation/utils_image.py
View File

@ -193,7 +193,7 @@ def mkdir_and_rename(path):
if os.path.exists(path): if os.path.exists(path):
new_name = path + '_archived_' + get_timestamp() new_name = path + '_archived_' + get_timestamp()
print('Path already exists. Rename it to [{:s}]'.format(new_name)) print('Path already exists. Rename it to [{:s}]'.format(new_name))
os.rename(path, new_name) os.replace(path, new_name)
os.makedirs(path) os.makedirs(path)

6
scripts/preload_models.py
View File

@ -260,7 +260,7 @@ def migrate_models_ckpt():
rename = yes_or_no(f'Ok to rename it to "{new_name}" for future reference?') rename = yes_or_no(f'Ok to rename it to "{new_name}" for future reference?')
if rename: if rename:
print(f'model.ckpt => {new_name}') print(f'model.ckpt => {new_name}')
os.rename(os.path.join(Model_dir,'model.ckpt'),os.path.join(Model_dir,new_name)) os.replace(os.path.join(Model_dir,'model.ckpt'),os.path.join(Model_dir,new_name))
#--------------------------------------------- #---------------------------------------------
def download_weight_datasets(models:dict, access_token:str): def download_weight_datasets(models:dict, access_token:str):
@ -347,12 +347,12 @@ def update_config_file(successfully_downloaded:dict,opt:dict):
try: try:
if os.path.exists(Config_file): if os.path.exists(Config_file):
print(f'** {Config_file} exists. Renaming to {Config_file}.orig') print(f'** {Config_file} exists. Renaming to {Config_file}.orig')
os.rename(Config_file,f'{Config_file}.orig') os.replace(Config_file,f'{Config_file}.orig')
tmpfile = os.path.join(os.path.dirname(Config_file),'new_config.tmp') tmpfile = os.path.join(os.path.dirname(Config_file),'new_config.tmp')
with open(tmpfile, 'w') as outfile: with open(tmpfile, 'w') as outfile:
outfile.write(Config_preamble) outfile.write(Config_preamble)
outfile.write(yaml) outfile.write(yaml)
os.rename(tmpfile,Config_file) os.replace(tmpfile,Config_file)
except Exception as e: except Exception as e:
print(f'**Error creating config file {Config_file}: {str(e)} **') print(f'**Error creating config file {Config_file}: {str(e)} **')