Fix clip path in migrate script

This commit is contained in:
Sergey Borisov 2023-07-05 13:43:09 +03:00
parent 0981a7d049
commit e3fc1b3816

View File

@ -219,11 +219,11 @@ class MigrateTo3(object):
repo_id = 'openai/clip-vit-large-patch14'
self._migrate_pretrained(CLIPTokenizer,
repo_id= repo_id,
dest= target_dir / 'clip-vit-large-patch14' / 'tokenizer',
dest= target_dir / 'clip-vit-large-patch14',
**kwargs)
self._migrate_pretrained(CLIPTextModel,
repo_id = repo_id,
dest = target_dir / 'clip-vit-large-patch14' / 'text_encoder',
dest = target_dir / 'clip-vit-large-patch14',
**kwargs)
# sd-2