Update app.py

app.py

@@ -33,7 +33,7 @@ import gradio as gr
 from lxml.html import fromstring
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file, save_file
-from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
+from diffusers import DiffusionPipeline, AutoencoderKL
 from PIL import Image, ImageDraw, ImageFont
 from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
 from refiners.fluxion.utils import manual_seed
@@ -440,7 +440,7 @@ MAX_SEED = np.iinfo(np.int32).max
 
 seq=512
 image_steps=50
-img_accu=9.0
+img_accu=7.0
 
 # ui data
 
@@ -500,7 +500,7 @@ function custom(){
 
 # torch pipes
 
-taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
+taef1 = AutoencoderKL.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
 good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=dtype).to(device)
 image_pipe = DiffusionPipeline.from_pretrained("ostris/Flex.1-alpha", torch_dtype=dtype, vae=taef1).to(device)
 image_pipe.enable_model_cpu_offload()