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

fix checkpoint VAE handling in migrate script

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-07 09:34:42 -04:00
parent 9f58ed35cf
commit 56f4712814
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/install/migrate_to_3.py
View File

@ -293,6 +293,7 @@ class MigrateTo3(object):
elif repo_id := vae.get('repo_id'): elif repo_id := vae.get('repo_id'):
if repo_id=='stabilityai/sd-vae-ft-mse': # this guy is already downloaded if repo_id=='stabilityai/sd-vae-ft-mse': # this guy is already downloaded
vae_path = 'models/core/convert/sd-vae-ft-mse' vae_path = 'models/core/convert/sd-vae-ft-mse'
return vae_path
else: else:
vae_path = self._download_vae(repo_id, vae.get('subfolder')) vae_path = self._download_vae(repo_id, vae.get('subfolder'))
@ -305,7 +306,10 @@ class MigrateTo3(object):
info = ModelProbe().heuristic_probe(vae_path) info = ModelProbe().heuristic_probe(vae_path)
dest = self._model_probe_to_path(info) / vae_path.name dest = self._model_probe_to_path(info) / vae_path.name
if not dest.exists(): if not dest.exists():
self.copy_dir(vae_path,dest) if vae_path.is_dir():
self.copy_dir(vae_path,dest)
else:
self.copy_file(vae_path,dest)
vae_path = dest vae_path = dest
if vae_path.is_relative_to(self.dest_models): if vae_path.is_relative_to(self.dest_models):