Skip to content

mlx_nested_unet Implementation#57

Open
ethanernst11 wants to merge 6 commits intoapple:mainfrom
ethanernst11:ethan/mlx
Open

mlx_nested_unet Implementation#57
ethanernst11 wants to merge 6 commits intoapple:mainfrom
ethanernst11:ethan/mlx

Conversation

@ethanernst11
Copy link
Contributor

Need to add Bella's mlx_unet when ready. Also need to test the file before merging.

@luke-carlson
Copy link
Contributor

take a look at #53 for an example on how to create a test case that compares mlx version vs pytorch version and checks that the output is the same

@ethanernst11 ethanernst11 marked this pull request as ready for review December 21, 2024 16:16
@ethanernst11
Copy link
Contributor Author

I figured I would open the PR for now and I will add the tests as soon as I am able to.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants