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

Fix import order of normalizing_flow (#326)

Browse Source
This commit is contained in:
AtomicVar
2024-01-17 00:45:55 +08:00
committed by GitHub
parent 2287294723
commit ce7b65e8c4
3 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
normalizing_flow/flows.py
View File

@@ -1,11 +1,10 @@
# Copyright © 2023-2024 Apple Inc.
from typing import Tuple, Optional, Union
from typing import Optional, Tuple, Union
import mlx.core as mx
import mlx.nn as nn
from bijectors import MaskedCoupling, AffineBijector
from bijectors import AffineBijector, MaskedCoupling
from distributions import Normal