QJerry multimodalart HF Staff commited on
Commit
56e0141
·
verified ·
1 Parent(s): 960765b

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>

Files changed (2) hide show
  1. app.py +2 -1
  2. 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