monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify import paths for controlnet file

Browse Source
This commit is contained in:
filipstrand 2024-09-21 10:23:05 +02:00
parent e007402cfa
commit 5a7157e7e8
3 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/mflux/__init__.py
View File

@ -1,5 +1,8 @@
from mflux.flux.flux import Flux1
from mflux.config.config import Config
from mflux.config.config import ConfigControlnet
from mflux.config.model_config import ModelConfig
from mflux.controlnet.flux_controlnet import Flux1Controlnet
from mflux.flux.flux import Flux1
from mflux.post_processing.image_util import ImageUtil
__all__ = ["Flux1", "Config", "ModelConfig"]
__all__ = ["Flux1", "Flux1Controlnet", "Config", "ConfigControlnet", "ModelConfig", "ImageUtil"]

2
src/mflux/generate.py
View File

@ -1,6 +1,4 @@
import argparse
import os
import sys
import time
from mflux import Flux1, Config, ModelConfig

9
src/mflux/generate_controlnet.py
View File

@ -1,14 +1,7 @@
import argparse
import os
import sys
import time
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
from mflux.config.model_config import ModelConfig
from mflux.config.config import ConfigControlnet
from mflux.controlnet.flux_controlnet import Flux1Controlnet
from mflux.post_processing.image_util import ImageUtil
from mflux import Flux1Controlnet, ConfigControlnet, ModelConfig, ImageUtil
def main():