From 4c5322ca856a250281e72bede89da61a8a33284a Mon Sep 17 00:00:00 2001 From: Kevin Black Date: Mon, 26 Jun 2023 22:35:24 -0700 Subject: [PATCH] Device specific seed --- scripts/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train.py b/scripts/train.py index 80da2e6..4b58f74 100644 --- a/scripts/train.py +++ b/scripts/train.py @@ -45,7 +45,7 @@ def main(_): logger.info(config) # set seed - set_seed(config.seed) + set_seed(config.seed, device_specific=True) # load scheduler, tokenizer and models. pipeline = StableDiffusionPipeline.from_pretrained(config.pretrained.model, revision=config.pretrained.revision)