Spaces:
Running
on
L4
Running
on
L4
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 |
-
|
| 469 |
-
|
|
|
|
| 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."
|