reverting back to DiffusionPipeline

app.py

@@ -22,7 +22,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 torch.cuda.max_memory_allocated(device=device)
 
-pipe1 = StableDiffusionXLPipeline.from_pretrained("FFusion/FFbasetest", torch_dtype=torch.float16, use_safetensors=True)
+pipe1 = DiffusionPipeline.from_pretrained("FFusion/FFbasetest", torch_dtype=torch.float16, use_safetensors=True)
 pipe2 = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
 
 pipe1 = pipe1.to(device)
@@ -80,8 +80,8 @@ gr.Interface(fn=genie,
                      gr.Textbox(label='Things the AI should not create (negative prompt)'), 
                      gr.Slider(1, 15, 10), gr.Slider(25, maximum=100, value=50, step=1), 
                      gr.Slider(minimum=1, step=1, maximum=999999999999999999, randomize=True)], 
-             outputs=[gc.Image(type='numpy', label="Generated Image"), gc.Image(type='numpy', label="Refined Image")],
-             title="FFusionXL - Generate and Refine", 
+             outputs=[gc.Image(type='numpy', label="FFusionXL Base Image"), gc.Image(type='numpy', label="Refined Image")],
+             title="FFusionXL Base - Generate and Refine", 
              description='<div style="display: flex; flex-wrap: wrap; gap: 2px;"><a href="https://huggingface.co/FFusion/FFusionXL-BASE" target="_new" rel="ugc"><img src="https://img.shields.io/badge/FFusionXL--BASE--SDXL-Model-pink" alt="FFusionXL-BASE-SDXL"></a>  <a href="https://huggingface.co/FFusion/FFusionXL-09-SDXL/blob/main/LICENSE.md" target="_new" rel="ugc"><img src="https://img.shields.io/badge/License-FFXL%20Research%20License-blue"></a></div>', 
              article = '**Citation** \
                         Please note that the demo is intended for academic and research purposes ONLY. Any use of the demo for generating inappropriate content is strictly prohibited. The responsibility for any misuse or inappropriate use of the demo lies solely with the users who generated such content, and this demo shall not be held liable for any such use. Original code: Manjushri. By interacting within this environment, you hereby acknowledge and agree to the terms of the SDXL 0.9 Research License. \