Plat commited on
Commit
31c0e08
·
1 Parent(s): 31cd180

feat: update deps, add zero gpu progress

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -1
app.py CHANGED
@@ -55,6 +55,7 @@ def on_generate(
55
  seed: int,
56
  randomize_seed: bool = True,
57
  num_images: int = 4,
 
58
  ):
59
  if image is not None:
60
  image = image.convert("RGB")
 
55
  seed: int,
56
  randomize_seed: bool = True,
57
  num_images: int = 4,
58
+ progress=gr.Progress(track_tqdm=True),
59
  ):
60
  if image is not None:
61
  image = image.convert("RGB")
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- git+https://github.com/p1atdev/vision-ft@94f17aae4
2
  accelerate
3
  diffusers
4
  transformers
 
1
+ git+https://github.com/p1atdev/vision-ft@ac60b7447
2
  accelerate
3
  diffusers
4
  transformers