diff --git a/README.md b/README.md index bed2eb2..6f84615 100644 --- a/README.md +++ b/README.md @@ -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)