Device specific seed

2023-06-26 22:35:24 -07:00
parent 1ce0994c8a
commit 4c5322ca85
1 changed files with 1 additions and 1 deletions
--- 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)