zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-10-24 22:58:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: stable-diffusion t2i add --seed (#558)

Browse Source
This commit is contained in:
zweifisch
2024-03-10 21:12:54 +08:00
committed by GitHub
parent ad3cf5ed98
commit d0fa6cfcae
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
stable_diffusion/stable_diffusion/__init__.py
View File

@@ -110,7 +110,7 @@ class StableDiffusion:
seed=None,
):
# Set the PRNG state
seed = seed or int(time.time())
seed = int(time.time()) if seed is None else seed
mx.random.seed(seed)
# Get the text conditioning
@@ -140,7 +140,7 @@ class StableDiffusion:
seed=None,
):
# Set the PRNG state
seed = seed or int(time.time())
seed = int(time.time()) if seed is None else seed
mx.random.seed(seed)
# Define the num steps and start step
@@ -238,7 +238,7 @@ class StableDiffusionXL(StableDiffusion):
seed=None,
):
# Set the PRNG state
seed = seed or int(time.time())
seed = int(time.time()) if seed is None else seed
mx.random.seed(seed)
# Get the text conditioning