from diffsynth_engine import fetch_model, WanVideoPipeline, WanPipelineConfig
from diffsynth_engine.utils.video import save_video
config = WanPipelineConfig.basic_config(
    model_path=fetch_model(
        "Wan-AI/Wan2.2-TI2V-5B",
        revision="bf16",
        path=[
            "diffusion_pytorch_model-00001-of-00003-bf16.safetensors",
            "diffusion_pytorch_model-00002-of-00003-bf16.safetensors",
            "diffusion_pytorch_model-00003-of-00003-bf16.safetensors",
        ],
    ),
    parallelism=1,
    offload_mode=None,
)
pipe = WanVideoPipeline.from_pretrained(config)
image = Image.open("input/wan_i2v_input.jpg").convert("RGB")
video = pipe(
    prompt="Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside.",
    negative_prompt="",
    input_image=image,
    num_frames=121,
    width=704,
    height=1280,
    seed=42,
)
save_video(video, "wan_ti2v.mp4", fps=pipe.config.fps)

from diffsynth_engine import fetch_model, FluxImagePipeline, FluxModelConfig
model_path = fetch_model("muse/flux-with-vae", path="flux1-dev-with-vae.safetensors")
config = FluxModelConfig(dit_path=model_path, use_fp8_linear=True, use_fsdp=True)
pipe = FluxImagePipeline.from_pretrained(config, parallelism=8, use_cfg_parallel=True)

from diffsynth_engine import fetch_model, FluxImagePipeline, FluxPipelineConfig
model_path = fetch_model("muse/flux-with-vae", path="flux1-dev-with-vae.safetensors")
config = FluxPipelineConfig(
    model_path=model_path,
    use_fp8_linear=True,
    parallelism=8,
    use_cfg_parallel=True,
    use_fsdp=True,
)
pipe = FluxImagePipeline.from_pretrained(config)

from diffsynth_engine import fetch_model, FluxImagePipeline, FluxPipelineConfig
model_path = fetch_model("muse/flux-with-vae", path="flux1-dev-with-vae.safetensors")
config = FluxPipelineConfig.basic_config(model_path=model_path, parallelism=8)
pipe = FluxImagePipeline.from_pretrained(config)