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 StableDiffusionPipeline
+from diffusers import DiffusionPipeline
 from PIL import Image, ImageDraw, ImageFont
 from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
 from refiners.fluxion.utils import manual_seed
@@ -445,7 +445,7 @@ MAX_SEED = np.iinfo(np.int32).max
 # precision data
 
 seq=512
-image_steps=300
+image_steps=50
 img_accu=6.5
 
 # ui data
@@ -507,8 +507,8 @@ function custom(){
 
 # torch pipes
 
-image_pipe = StableDiffusionPipeline.from_pretrained("emilianJR/epiCRealism", torch_dtype=dtype).to(device)
-#image_pipe.enable_model_cpu_offload()
+image_pipe = DiffusionPipeline.from_pretrained("ostris/Flex.1-alpha", torch_dtype=dtype).to(device)
+image_pipe.enable_model_cpu_offload()
 
 torch.cuda.empty_cache()