FLUX: rename flux to mlx_flux

2025-08-29 18:26:37 +08:00 · 2024-11-07 10:32:53 +08:00 · 2024-11-07 10:32:53 +08:00 · 83c92c2a11
commit 83c92c2a11
parent ed9e81dd58
15 changed files with 2 additions and 2 deletions
--- a/flux/dreambooth.py
+++ b/flux/dreambooth.py
@ -13,7 +13,7 @@ from mlx.nn.utils import average_gradients
 from mlx.utils import tree_flatten, tree_map, tree_reduce
 from PIL import Image

-from flux import FluxPipeline, Trainer, load_dataset
+from mlx_flux import FluxPipeline, Trainer, load_dataset


 def generate_progress_images(iteration, flux, args):
--- a/flux/mlx_flux/init.py
+++ b/flux/mlx_flux/init.py
--- a/flux/mlx_flux/autoencoder.py
+++ b/flux/mlx_flux/autoencoder.py
--- a/flux/mlx_flux/clip.py
+++ b/flux/mlx_flux/clip.py
--- a/flux/mlx_flux/datasets.py
+++ b/flux/mlx_flux/datasets.py
--- a/flux/mlx_flux/flux.py
+++ b/flux/mlx_flux/flux.py
--- a/flux/mlx_flux/layers.py
+++ b/flux/mlx_flux/layers.py
--- a/flux/mlx_flux/lora.py
+++ b/flux/mlx_flux/lora.py
--- a/flux/mlx_flux/model.py
+++ b/flux/mlx_flux/model.py
--- a/flux/mlx_flux/sampler.py
+++ b/flux/mlx_flux/sampler.py
--- a/flux/mlx_flux/t5.py
+++ b/flux/mlx_flux/t5.py
--- a/flux/mlx_flux/tokenizers.py
+++ b/flux/mlx_flux/tokenizers.py
--- a/flux/mlx_flux/trainer.py
+++ b/flux/mlx_flux/trainer.py
--- a/flux/mlx_flux/utils.py
+++ b/flux/mlx_flux/utils.py
--- a/flux/txt2image.py
+++ b/flux/txt2image.py
@ -8,7 +8,7 @@ import numpy as np
 from PIL import Image
 from tqdm import tqdm

-from flux import FluxPipeline
+from mlx_flux import FluxPipeline


 def to_latent_size(image_size):