Spaces:
Running
on
Zero
Running
on
Zero
Add FA3 and some other enhancements (#1)
Browse files- Update requirements.txt (28aa1ce52ade1b4c3f5ca2f4984bbfba96cae32f)
- Update app.py (3bddc9bf567961e5be63a7c4609def4fdbdf2d6e)
Co-authored-by: Apolinário from multimodal AI art <multimodalart@users.noreply.huggingface.co>
- app.py +2 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -190,6 +190,7 @@ def generate_image(
|
|
| 190 |
num_inference_steps=50,
|
| 191 |
shift=3.0,
|
| 192 |
max_sequence_length=512,
|
|
|
|
| 193 |
):
|
| 194 |
height, width = get_resolution(resolution)
|
| 195 |
|
|
@@ -377,7 +378,7 @@ def prompt_enhance(prompt, enable_enhance):
|
|
| 377 |
|
| 378 |
|
| 379 |
@spaces.GPU
|
| 380 |
-
def generate(prompt, resolution, seed, steps, shift, enhance, random_seed, gallery_images):
|
| 381 |
if pipe is None:
|
| 382 |
raise gr.Error("Model not loaded.")
|
| 383 |
|
|
|
|
| 190 |
num_inference_steps=50,
|
| 191 |
shift=3.0,
|
| 192 |
max_sequence_length=512,
|
| 193 |
+
progress=gr.Progress(track_tqdm=True)
|
| 194 |
):
|
| 195 |
height, width = get_resolution(resolution)
|
| 196 |
|
|
|
|
| 378 |
|
| 379 |
|
| 380 |
@spaces.GPU
|
| 381 |
+
def generate(prompt, resolution, seed, steps, shift, enhance, random_seed, gallery_images, progress=gr.Progress(track_tqdm=True)):
|
| 382 |
if pipe is None:
|
| 383 |
raise gr.Error("Model not loaded.")
|
| 384 |
|
requirements.txt
CHANGED
|
@@ -5,3 +5,4 @@ accelerate
|
|
| 5 |
spaces
|
| 6 |
openai
|
| 7 |
git+https://github.com/JerryWu-code/diffusers.git@c3a2d3c5e570ee53fc529e274871ebc98a433ddd
|
|
|
|
|
|
| 5 |
spaces
|
| 6 |
openai
|
| 7 |
git+https://github.com/JerryWu-code/diffusers.git@c3a2d3c5e570ee53fc529e274871ebc98a433ddd
|
| 8 |
+
kernels
|