Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes configure_invokeai.py crash on Windows systems

Browse Source 
The step in which the new models.yaml file replaces the old one was
crashing on Windows due to the fact that on Windows, the os.rename()
function will refuse to replace an existing file, unlike the behavior
on Linux and Mac. The os.replace() function, which was introduced in
python3, supposedly fixes this.
This commit is contained in:
Lincoln Stein 2022-11-21 07:25:04 +00:00
parent 0661256b61
commit 3c5a14a814
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/configure_invokeai.py
View File

@ -232,7 +232,7 @@ def migrate_models_ckpt():
rename = yes_or_no(f'Ok to rename it to "{new_name}" for future reference?')
if rename:
print(f'model.ckpt => {new_name}')
os.rename(os.path.join(model_path,'model.ckpt'),os.path.join(model_path,new_name))
os.replace(os.path.join(model_path,'model.ckpt'),os.path.join(model_path,new_name))
#---------------------------------------------
def download_weight_datasets(models:dict, access_token:str):
@ -340,12 +340,12 @@ def update_config_file(successfully_downloaded:dict,opt:dict):
try:
if os.path.exists(config_file):
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')
with open(tmpfile, 'w') as outfile:
outfile.write(Config_preamble)
outfile.write(yaml)
os.rename(tmpfile,config_file)
os.replace(tmpfile,config_file)
except Exception as e:
print(f'**Error creating config file {config_file}: {str(e)} **')