Update readme with dev model description

This commit is contained in:
filipstrand 2024-08-20 21:50:37 +02:00
parent f9f032a220
commit 01c8a372e6

View File

@ -37,19 +37,27 @@ like [Numpy](https://numpy.org) and [Pillow](https://pypi.org/project/pillow/) f
```
### Generating an image
Run the provided [main.py](main.py) by specifying a prompt and some optional arguments like so:
Run the provided [main.py](main.py) by specifying a prompt and some optional arguments like so using the default `Schnell` model:
```
python main.py --prompt "Luxury food photograph" --steps 2 --seed 2
```
or use the slower, but more powerful `Dev` model and run it with more time steps:
```
python main.py --model dev --prompt "Luxury food photograph" --steps 25 --seed 2
```
⚠️ *If the specific model is not already downloaded on your machine, it will start the download process and fetch the model weights (~34GB in size for the Schnell or Dev model respectively).* ⚠️
#### Full list of Command-Line Arguments
- **`--prompt`** (required, `str`): Text description of the image to generate.
- **`--output`** (optional, `str`, default: `"image.png"`): Output image filename.
- **`--model`** (optional, `str`, default: `"schnell"`): Model to use for generation.
- **`--model`** (optional, `str`, default: `"schnell"`): Model to use for generation (`"schnell"` or `"dev"`).
- **`--seed`** (optional, `int`, default: `0`): Seed for random number generation. Default is time-based.
@ -85,8 +93,6 @@ image = flux.generate_image(
ImageUtil.save_image(image, "image.png")
```
If the model is not already downloaded on your machine, it will start the download process and fetch the model weights (~34GB in size for the Schnell model).
### Image generation speed (updated)