refine examples

examples/flux/FLUX.1-dev-IP-Adapter.py

@@ -1,10 +1,7 @@
 import torch
-from PIL import Image
-from diffsynth import save_video, VideoData, download_models
 from diffsynth.pipelines.flux_image_new import FluxImagePipeline, ModelConfig
-from modelscope import dataset_snapshot_download
 
-#TODO: repalce the local path with model_id
+
 pipe = FluxImagePipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
     device="cuda",
@@ -14,16 +11,14 @@ pipe = FluxImagePipeline.from_pretrained(
         ModelConfig(model_id="black-forest-labs/FLUX.1-dev", origin_file_pattern="text_encoder_2/"),
         ModelConfig(model_id="black-forest-labs/FLUX.1-dev", origin_file_pattern="ae.safetensors"),
         ModelConfig(model_id="InstantX/FLUX.1-dev-IP-Adapter", origin_file_pattern="ip-adapter.bin"),
-        ModelConfig(path="models/IpAdapter/InstantX/FLUX.1-dev-IP-Adapter/image_encoder")
+        ModelConfig(model_id="google/siglip-so400m-patch14-384"),
     ],
 )
 
-seed = 42
 origin_prompt = "a rabbit in a garden, colorful flowers"
-image = pipe(prompt=origin_prompt, height=1280, width=960, seed=seed)
+image = pipe(prompt=origin_prompt, height=1280, width=960, seed=42)
 image.save("style image.jpg")
 
-torch.manual_seed(seed)
-image = pipe(prompt="A piggy", height=1280, width=960, seed=seed,
+image = pipe(prompt="A piggy", height=1280, width=960, seed=42,
     ipadapter_images=[image], ipadapter_scale=0.7)
 image.save("A piggy.jpg")

examples/flux/Step1X-Edit.py

@@ -1,13 +1,9 @@
 import torch
 from diffsynth.pipelines.flux_image_new import FluxImagePipeline, ModelConfig
-from modelscope import snapshot_download
 from PIL import Image
 import numpy as np
 
 
-snapshot_download("Qwen/Qwen2.5-VL-7B-Instruct", cache_dir="./models")
-snapshot_download("stepfun-ai/Step1X-Edit", cache_dir="./models")
-
 pipe = FluxImagePipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
     device="cuda",
@@ -18,27 +14,19 @@ pipe = FluxImagePipeline.from_pretrained(
     ],
 )
 
-
-pipe.enable_vram_management()
-
 image = Image.fromarray(np.zeros((1248, 832, 3), dtype=np.uint8) + 255)
 image = pipe(
     prompt="draw red flowers in Chinese ink painting style",
     step1x_reference_image=image,
     width=832, height=1248, cfg_scale=6,
-    seed=1,
-    rand_device='cuda'
+    seed=1, rand_device='cuda'
 )
 image.save("image_1.jpg")
 
-
-
 image = pipe(
     prompt="add more flowers in Chinese ink painting style",
     step1x_reference_image=image,
     width=832, height=1248, cfg_scale=6,
-    seed=2,
-    rand_device='cuda'
+    seed=2, rand_device='cuda'
 )
 image.save("image_2.jpg")
-