from lightx2v import LightX2VPipeline
# Initialize pipelinepipe = LightX2VPipeline( model_path="Tongyi-MAI/Z-Image-Turbo", model_cls="z_image", task="t2i",)
# Step 1: Enable quantization (FP8 transformer + INT4 text encoder)pipe.enable_quantize( dit_quantized=True, dit_quantized_ckpt="lightx2v/Z-Image-Turbo-Quantized/z_image_turbo_scaled_fp8_e4m3fn.safetensors", quant_scheme="fp8-sgl", # IMPORTANT: Use int4 Qwen3 for 8GB VRAM text_encoder_quantized=True, text_encoder_quantized_ckpt="JunHowie/Qwen3-4B-GPTQ-Int4", text_encoder_quant_scheme="int4")
# Step 2: Enable CPU offloadingpipe.enable_offload( cpu_offload=True, offload_granularity="model", # Use "model" for maximum memory savings)
# Step 3: Create generatorpipe.create_generator( attn_mode="flash_attn3", aspect_ratio="16:9", infer_steps=9, guidance_scale=1,)
# Step 4: Generate imagepipe.generate( seed=42, prompt="A beautiful landscape with mountains and lakes, ultra HD, 4K", negative_prompt="", save_result_path="output.png",)