diff --git a/examples/flux2/model_inference/FLUX.2-dev.py b/examples/flux2/model_inference/FLUX.2-dev.py index 4f5176c..c297536 100644 --- a/examples/flux2/model_inference/FLUX.2-dev.py +++ b/examples/flux2/model_inference/FLUX.2-dev.py @@ -1,6 +1,6 @@ from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig import torch - +from PIL import Image vram_config = { "offload_dtype": torch.bfloat16, @@ -25,3 +25,8 @@ pipe = Flux2ImagePipeline.from_pretrained( prompt = "Realistic macro photograph of a hermit crab using a soda can as its shell, partially emerging from the can, captured with sharp detail and natural colors, on a sunlit beach with soft shadows and a shallow depth of field, with blurred ocean waves in the background. The can has the text `BFL Diffusers` on it and it has a color gradient that start with #FF5733 at the top and transitions to #33FF57 at the bottom." image = pipe(prompt, seed=42, rand_device="cuda", num_inference_steps=50) image.save("image_FLUX.2-dev.jpg") + +prompt = "Transform the image into Japanese anime style" +edit_image = [Image.open("image_FLUX.2-dev.jpg")] +image = pipe(prompt, seed=42, rand_device="cuda", edit_image=edit_image, num_inference_steps=50, embedded_guidance=2.5) +image.save("image_FLUX.2-dev_edit.jpg") \ No newline at end of file diff --git a/examples/flux2/model_inference_low_vram/FLUX.2-dev.py b/examples/flux2/model_inference_low_vram/FLUX.2-dev.py index 41d588e..00428b1 100644 --- a/examples/flux2/model_inference_low_vram/FLUX.2-dev.py +++ b/examples/flux2/model_inference_low_vram/FLUX.2-dev.py @@ -1,5 +1,6 @@ from diffsynth.pipelines.flux2_image import Flux2ImagePipeline, ModelConfig import torch +from PIL import Image vram_config = { "offload_dtype": "disk", @@ -24,4 +25,9 @@ pipe = Flux2ImagePipeline.from_pretrained( ) prompt = "High resolution. A dreamy underwater portrait of a serene young woman in a flowing blue dress. Her hair floats softly around her face, strands delicately suspended in the water. Clear, shimmering light filters through, casting gentle highlights, while tiny bubbles rise around her. Her expression is calm, her features finely detailed—creating a tranquil, ethereal scene." image = pipe(prompt, seed=42, rand_device="cuda", num_inference_steps=50) -image.save("image.jpg") \ No newline at end of file +image.save("image.jpg") + +prompt = "Transform the image into Japanese anime style" +edit_image = [Image.open("image.jpg")] +image = pipe(prompt, seed=42, rand_device="cuda", edit_image=edit_image, num_inference_steps=50, embedded_guidance=2.5) +image.save("image_edit.jpg") \ No newline at end of file