minakshi.mathpal commited on
Commit
0e01082
·
1 Parent(s): 3e5cf09

refactored app.py

Browse files
Files changed (1) hide show
  1. custom_stable_diffusion.py +1 -1
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)