Add helpful hints when configuring which model to use in example

This commit is contained in:
filipstrand 2024-08-20 22:46:31 +02:00
parent 7a277fc326
commit 04c2aaac26

View File

@ -78,13 +78,13 @@ from flux_1.config.config import Config
from flux_1.flux import Flux1 from flux_1.flux import Flux1
from flux_1.post_processing.image_util import ImageUtil from flux_1.post_processing.image_util import ImageUtil
flux = Flux1.from_repo("black-forest-labs/FLUX.1-schnell") flux = Flux1.from_alias("schnell") # "schnell" or "dev"
image = flux.generate_image( image = flux.generate_image(
seed=3, seed=3,
prompt="Luxury food photograph of a birthday cake. In the middle it has three candles shaped like letters spelling the word 'MLX'. It has perfect lighting and a cozy background with big bokeh and shallow depth of field. The mood is a sunset balcony in tuscany. The photo is taken from the side of the cake. The scene is complemented by a warm, inviting light that highlights the textures and colors of the ingredients, giving it an appetizing and elegant look.", prompt="Luxury food photograph of a birthday cake. In the middle it has three candles shaped like letters spelling the word 'MLX'. It has perfect lighting and a cozy background with big bokeh and shallow depth of field. The mood is a sunset balcony in tuscany. The photo is taken from the side of the cake. The scene is complemented by a warm, inviting light that highlights the textures and colors of the ingredients, giving it an appetizing and elegant look.",
config=Config( config=Config(
num_inference_steps=2, num_inference_steps=2, # Schnell works well with 2-4 steps, Dev works well with 20-25 steps
height=768, height=768,
width=1360, width=1360,
) )