Add FA3 and some other enhancements

app.py

@@ -190,6 +190,7 @@ def generate_image(
     num_inference_steps=50,
     shift=3.0,
     max_sequence_length=512,
+    progress=gr.Progress(track_tqdm=True)
 ):
     height, width = get_resolution(resolution)
 
@@ -377,7 +378,7 @@ def prompt_enhance(prompt, enable_enhance):
 
 
 @spaces.GPU
-def generate(prompt, resolution, seed, steps, shift, enhance, random_seed, gallery_images):
+def generate(prompt, resolution, seed, steps, shift, enhance, random_seed, gallery_images, progress=gr.Progress(track_tqdm=True)):
     if pipe is None:
         raise gr.Error("Model not loaded.")
 

requirements.txt

@@ -5,3 +5,4 @@ accelerate
 spaces
 openai
 git+https://github.com/JerryWu-code/diffusers.git@c3a2d3c5e570ee53fc529e274871ebc98a433ddd
+kernels