LiuZichen commited on
Commit
99f78d1
·
verified ·
1 Parent(s): 2f06c4e

Update MagicQuill/brushnet/powerpaint_utils.py

Browse files
MagicQuill/brushnet/powerpaint_utils.py CHANGED
@@ -465,8 +465,9 @@ def add_tokens(
465
 
466
  # text_encoder.set_embedding_layer()
467
  embedding_layer = text_encoder.text_model.embeddings.token_embedding
468
- text_encoder.text_model.embeddings.token_embedding = EmbeddingLayerWithFixes(embedding_layer)
469
- embedding_layer = text_encoder.text_model.embeddings.token_embedding
 
470
 
471
  assert embedding_layer is not None, (
472
  "Do not support get embedding layer for current text encoder. " "Please check your configuration."
 
465
 
466
  # text_encoder.set_embedding_layer()
467
  embedding_layer = text_encoder.text_model.embeddings.token_embedding
468
+ if not isinstance(embedding_layer, EmbeddingLayerWithFixes):
469
+ text_encoder.text_model.embeddings.token_embedding = EmbeddingLayerWithFixes(embedding_layer)
470
+ embedding_layer = text_encoder.text_model.embeddings.token_embedding
471
 
472
  assert embedding_layer is not None, (
473
  "Do not support get embedding layer for current text encoder. " "Please check your configuration."