minakshi.mathpal commited on
Commit ·
0e01082
1
Parent(s): 3e5cf09
refactored app.py
Browse files
custom_stable_diffusion.py
CHANGED
|
@@ -262,7 +262,7 @@ class TextEmbeddingProcessor:
|
|
| 262 |
input_ids, position = self.tokenize_text(self.prompt)
|
| 263 |
token_embeddings = self.token_emb_layer(input_ids)
|
| 264 |
embeddings = self.load_embedding(concept_name)
|
| 265 |
-
|
| 266 |
if embeddings is not None:
|
| 267 |
# embeddings = embeddings.to(self.config.device)
|
| 268 |
replacement_token_embedding = embeddings[next(iter(embeddings.keys()))].to(self.config.device)
|
|
|
|
| 262 |
input_ids, position = self.tokenize_text(self.prompt)
|
| 263 |
token_embeddings = self.token_emb_layer(input_ids)
|
| 264 |
embeddings = self.load_embedding(concept_name)
|
| 265 |
+
print(embeddings)
|
| 266 |
if embeddings is not None:
|
| 267 |
# embeddings = embeddings.to(self.config.device)
|
| 268 |
replacement_token_embedding = embeddings[next(iter(embeddings.keys()))].to(self.config.device)
|