fix main
This commit is contained in:
parent
16ee26c2e9
commit
45309d0129
@ -20,7 +20,7 @@ like [Numpy](https://numpy.org) and [Pillow](https://pypi.org/project/pillow/) f
|
||||
### Models
|
||||
|
||||
- [x] FLUX.1-Scnhell
|
||||
- [ ] FLUX.1-Dev
|
||||
- [x] FLUX.1-Dev
|
||||
|
||||
### Installation
|
||||
1. Clone the repo:
|
||||
|
||||
4
main.py
4
main.py
@ -4,10 +4,10 @@ import sys
|
||||
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'src')))
|
||||
|
||||
from flux_1_schnell.config.config import Config
|
||||
from flux_1_schnell.flux import Flux1Schnell
|
||||
from flux_1_schnell.flux import Flux1
|
||||
from flux_1_schnell.post_processing.image_util import ImageUtil
|
||||
|
||||
flux = Flux1Schnell("black-forest-labs/FLUX.1-schnell")
|
||||
flux = Flux1("black-forest-labs/FLUX.1-schnell", max_sequence_length=256)
|
||||
|
||||
image = flux.generate_image(
|
||||
seed=3,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user