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

use format "diffusers" rather than format "folder" in models.yaml

Browse Source
This commit is contained in:
Lincoln Stein 2023-06-13 01:43:05 -04:00
parent 87ba17a1f5
commit a6af7e8824
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/migrate_models_to_3.0.py
View File

@ -185,7 +185,7 @@ def migrate_pipelines(dest_dir: Path, dest_yaml: io.TextIOBase):
'name': repo_name, 'name': repo_name,
'path': str(rel_path), 'path': str(rel_path),
'description': f'diffusers model {repo_id}', 'description': f'diffusers model {repo_id}',
'format': 'folder', 'format': 'diffusers',
'image_size': info.image_size, 'image_size': info.image_size,
'base': info.base_type.value, 'base': info.base_type.value,
'variant': info.variant_type.value, 'variant': info.variant_type.value,
@ -261,7 +261,11 @@ def main():
os.chdir(root_directory) os.chdir(root_directory)
with open(dest_yaml,'w') as yaml_file: with open(dest_yaml,'w') as yaml_file:
print(yaml.dump({'_version':'3.0.0'}),file=yaml_file,end="") print(yaml.dump({'__metadata__':
{'version':'3.0.0'}
}
),file=yaml_file,end=""
)
create_directory_structure(dest_directory) create_directory_structure(dest_directory)
migrate_support_models(dest_directory) migrate_support_models(dest_directory)
migrate_conversion_models(dest_directory) migrate_conversion_models(dest_directory)