github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-04 19:16:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixing insecure hash function

Browse Source
This commit is contained in:
Maruyama_Aya 2023-06-06 14:51:11 +08:00
parent b56c7f4283
commit 1c1f71cbd2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/tutorial/new_api/dreambooth/train_dreambooth_colossalai.py
View File

@ -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)