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

adjusted instructions for the released stable-diffusion-v1 weights

Browse Source
This commit is contained in:
Lincoln Stein
2022-08-22 15:33:27 -04:00
parent 9391eaff0e
commit a20827697c
4 changed files with 61 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm/modules/encoders/modules.py
View File

@ -146,8 +146,8 @@ class FrozenCLIPEmbedder(AbstractEncoder):
"""Uses the CLIP transformer encoder for text (from Hugging Face)"""
def __init__(self, version="openai/clip-vit-large-patch14", device="cuda", max_length=77):
super().__init__()
self.tokenizer = CLIPTokenizer.from_pretrained(version)
self.transformer = CLIPTextModel.from_pretrained(version)
self.tokenizer = CLIPTokenizer.from_pretrained(version,local_files_only=True)
self.transformer = CLIPTextModel.from_pretrained(version,local_files_only=True)
self.device = device
self.max_length = max_length
self.freeze()