zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 12:49:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

work with tuple shape (#393)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-01 13:03:47 -08:00
committed by GitHub
parent 0340113e02
commit ec14583c2a
6 changed files with 5 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
stable_diffusion/README.md
View File

@@ -48,8 +48,6 @@ latent_generator = sd.generate_latents("A photo of an astronaut riding a horse o
# Here we are evaluating each diffusion step but we could also evaluate
# once at the end.
for x_t in latent_generator:
mx.simplify(x_t) # remove possible redundant computation eg reuse
# scalars etc
mx.eval(x_t)
# Now x_t is the last latent from the reverse process aka x_0. We can