Update app.py

app.py

@@ -39,6 +39,7 @@ from tqdm import tqdm
 from einops import rearrange, repeat
 from torch import Tensor, nn
 from pipeline import FluxWithCFGPipeline
+from diffusers.models.autoencoders import AutoencoderKL
 from transformers import CLIPModel, CLIPProcessor, CLIPTextModel, CLIPTokenizer, CLIPConfig, T5EncoderModel, T5Tokenizer
 import gc
 import warnings