Update pipeline.py

pipeline.py

@@ -6,6 +6,8 @@ import sys
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import os
+from torchvision.utils import save_image
+from PIL import Image
 from .vae import AutoencoderKL
 from .mar import mar_base, mar_large, mar_huge