fixing insecure hash function

1c1f71cb · Maruyama_Aya · b56c7f42 · 1c1f71cb
Commit 1c1f71cb authored Jun 06, 2023 by Maruyama_Aya
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/tutorial/new_api/dreambooth/train_dreambooth_colossalai.py ...utorial/new_api/dreambooth/train_dreambooth_colossalai.py +1 -1

No files found.
--- a/examples/tutorial/new_api/dreambooth/train_dreambooth_colossalai.py
+++ b/examples/tutorial/new_api/dreambooth/train_dreambooth_colossalai.py
@@ -397,7 +397,7 @@ def main(args):
                images = pipeline(example["prompt"]).images
                for i, image in enumerate(images):
-                    hash_image = hashlib.sha1(image.tobytes()).hexdigest()
+                    hash_image = hashlib.sha256(image.tobytes()).hexdigest()
                    image_filename = class_images_dir / f"{example['index'][i] + cur_class_images}-{hash_image}.jpg"
                    image.save(image_filename)