lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow clip to skip TI

Browse Source
This commit is contained in:
layerdiffusion
2024-08-04 19:28:53 -07:00
parent 07a02f751d
commit e28e11fa97
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/text_processing/classic_engine.py
View File

@@ -49,14 +49,17 @@ class CLIPEmbeddingForTextualInversion(torch.nn.Module):
class ClassicTextProcessingEngine(torch.nn.Module):
def __init__(self, text_encoder, tokenizer, chunk_length=75,
embedding_dir='./embeddings', embedding_key='clip_l', embedding_expected_shape=768, emphasis_name="original",
embedding_dir=None, embedding_key='clip_l', embedding_expected_shape=768, emphasis_name="original",
text_projection=False, minimal_clip_skip=1, clip_skip=1, return_pooled=False, final_layer_norm=True,
callback_before_encode=None):
super().__init__()
self.embeddings = EmbeddingDatabase(tokenizer, embedding_expected_shape)
self.embeddings.add_embedding_dir(embedding_dir)
self.embeddings.load_textual_inversion_embeddings()
if isinstance(embedding_dir, str):
self.embeddings.add_embedding_dir(embedding_dir)
self.embeddings.load_textual_inversion_embeddings()
self.embedding_key = embedding_key
self.text_encoder = text_encoder