Fix import
This commit is contained in:
parent
8e5609132c
commit
c5f8b73d9d
@ -172,7 +172,7 @@ This is useful for integrating MFLUX into shell scripts or dynamically generatin
|
|||||||
Alternatively, you can use MFLUX directly in Python:
|
Alternatively, you can use MFLUX directly in Python:
|
||||||
|
|
||||||
```python
|
```python
|
||||||
from mflux import Flux1
|
from mflux.models.flux.variants.txt2img.flux import Flux1
|
||||||
|
|
||||||
# Load the model
|
# Load the model
|
||||||
flux = Flux1.from_name(
|
flux = Flux1.from_name(
|
||||||
|
|||||||
@ -4,8 +4,3 @@ import os
|
|||||||
# This must be set before any tokenizers are imported/used
|
# This must be set before any tokenizers are imported/used
|
||||||
if "TOKENIZERS_PARALLELISM" not in os.environ:
|
if "TOKENIZERS_PARALLELISM" not in os.environ:
|
||||||
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
||||||
|
|
||||||
# Export main classes for convenient import
|
|
||||||
from mflux.models.flux.variants.txt2img.flux import Flux1
|
|
||||||
|
|
||||||
__all__ = ["Flux1"]
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user