jadechoghari commited on
Commit
9ff7343
1 Parent(s): 9885231

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -0
pipeline.py CHANGED
@@ -7,6 +7,7 @@ from huggingface_hub import hf_hub_download
7
  from safetensors.torch import load_file
8
  import os
9
  from torchvision.utils import save_image
 
10
  from safetensors.torch import load_file
11
  from .vae import AutoencoderKL
12
  from .mar import mar_base, mar_large, mar_huge
 
7
  from safetensors.torch import load_file
8
  import os
9
  from torchvision.utils import save_image
10
+ from PIL import Image
11
  from safetensors.torch import load_file
12
  from .vae import AutoencoderKL
13
  from .mar import mar_base, mar_large, mar_huge