Skip to content

Commit

Permalink
df: log prompts in use
Browse files Browse the repository at this point in the history
  • Loading branch information
bghira committed Apr 29, 2024
1 parent 4bffd70 commit 150b2ba
Showing 1 changed file with 2 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -421,6 +421,8 @@ def __call__(self, **args):
# Grab prompt embeds from T5.
prompt = args.get("prompt", "")
negative_prompt = user_config.get("negative_prompt", "")
logging.debug(f"Positive prompt: {prompt}")
logging.debug(f"Negative prompt: {negative_prompt}")
embeds, parameters = self._embeds(
[prompt] * self.batch_size(), [negative_prompt] * self.batch_size()
)
Expand Down

0 comments on commit 150b2ba

Please sign in to comment.