From 83c92c2a113983bba1059249e6e7568c089d8788 Mon Sep 17 00:00:00 2001 From: madroid Date: Thu, 7 Nov 2024 10:32:53 +0800 Subject: [PATCH] FLUX: rename flux to mlx_flux --- flux/dreambooth.py | 2 +- flux/{flux => mlx_flux}/__init__.py | 0 flux/{flux => mlx_flux}/autoencoder.py | 0 flux/{flux => mlx_flux}/clip.py | 0 flux/{flux => mlx_flux}/datasets.py | 0 flux/{flux => mlx_flux}/flux.py | 0 flux/{flux => mlx_flux}/layers.py | 0 flux/{flux => mlx_flux}/lora.py | 0 flux/{flux => mlx_flux}/model.py | 0 flux/{flux => mlx_flux}/sampler.py | 0 flux/{flux => mlx_flux}/t5.py | 0 flux/{flux => mlx_flux}/tokenizers.py | 0 flux/{flux => mlx_flux}/trainer.py | 0 flux/{flux => mlx_flux}/utils.py | 0 flux/txt2image.py | 2 +- 15 files changed, 2 insertions(+), 2 deletions(-) rename flux/{flux => mlx_flux}/__init__.py (100%) rename flux/{flux => mlx_flux}/autoencoder.py (100%) rename flux/{flux => mlx_flux}/clip.py (100%) rename flux/{flux => mlx_flux}/datasets.py (100%) rename flux/{flux => mlx_flux}/flux.py (100%) rename flux/{flux => mlx_flux}/layers.py (100%) rename flux/{flux => mlx_flux}/lora.py (100%) rename flux/{flux => mlx_flux}/model.py (100%) rename flux/{flux => mlx_flux}/sampler.py (100%) rename flux/{flux => mlx_flux}/t5.py (100%) rename flux/{flux => mlx_flux}/tokenizers.py (100%) rename flux/{flux => mlx_flux}/trainer.py (100%) rename flux/{flux => mlx_flux}/utils.py (100%) diff --git a/flux/dreambooth.py b/flux/dreambooth.py index 48dcad47..9dcaffb3 100644 --- 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): diff --git a/flux/flux/__init__.py b/flux/mlx_flux/__init__.py similarity index 100% rename from flux/flux/__init__.py rename to flux/mlx_flux/__init__.py diff --git a/flux/flux/autoencoder.py b/flux/mlx_flux/autoencoder.py similarity index 100% rename from flux/flux/autoencoder.py rename to flux/mlx_flux/autoencoder.py diff --git a/flux/flux/clip.py b/flux/mlx_flux/clip.py similarity index 100% rename from flux/flux/clip.py rename to flux/mlx_flux/clip.py diff --git a/flux/flux/datasets.py b/flux/mlx_flux/datasets.py similarity index 100% rename from flux/flux/datasets.py rename to flux/mlx_flux/datasets.py diff --git a/flux/flux/flux.py b/flux/mlx_flux/flux.py similarity index 100% rename from flux/flux/flux.py rename to flux/mlx_flux/flux.py diff --git a/flux/flux/layers.py b/flux/mlx_flux/layers.py similarity index 100% rename from flux/flux/layers.py rename to flux/mlx_flux/layers.py diff --git a/flux/flux/lora.py b/flux/mlx_flux/lora.py similarity index 100% rename from flux/flux/lora.py rename to flux/mlx_flux/lora.py diff --git a/flux/flux/model.py b/flux/mlx_flux/model.py similarity index 100% rename from flux/flux/model.py rename to flux/mlx_flux/model.py diff --git a/flux/flux/sampler.py b/flux/mlx_flux/sampler.py similarity index 100% rename from flux/flux/sampler.py rename to flux/mlx_flux/sampler.py diff --git a/flux/flux/t5.py b/flux/mlx_flux/t5.py similarity index 100% rename from flux/flux/t5.py rename to flux/mlx_flux/t5.py diff --git a/flux/flux/tokenizers.py b/flux/mlx_flux/tokenizers.py similarity index 100% rename from flux/flux/tokenizers.py rename to flux/mlx_flux/tokenizers.py diff --git a/flux/flux/trainer.py b/flux/mlx_flux/trainer.py similarity index 100% rename from flux/flux/trainer.py rename to flux/mlx_flux/trainer.py diff --git a/flux/flux/utils.py b/flux/mlx_flux/utils.py similarity index 100% rename from flux/flux/utils.py rename to flux/mlx_flux/utils.py diff --git a/flux/txt2image.py b/flux/txt2image.py index 5ebec81a..fc209b17 100644 --- 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):