Skip to content

Commit

Permalink
Print seed when doing a prediction
Browse files Browse the repository at this point in the history
  • Loading branch information
jd7h committed Jan 16, 2024
1 parent 33dd61e commit d07aeaf
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions predict.py
Expand Up @@ -143,6 +143,7 @@ def predict(
# randomize seed if necessary
if seed is None:
seed = random.randint(0, MAX_SEED)
print(f"Using seed {seed}...")

# check the prompt for the trigger word
image_token_id = self.pipe.tokenizer.convert_tokens_to_ids(self.pipe.trigger_word)
Expand Down

0 comments on commit d07aeaf

Please sign in to comment.