Prgckwb commited on
Commit
1dcfe8d
1 Parent(s): 23f2de3

:tada: change some process

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -75,6 +75,7 @@ def inference(
75
 
76
  generator = torch.Generator(device=device).manual_seed(seed)
77
 
 
78
  images = pipe(
79
  prompt,
80
  negative_prompt=negative_prompt,
@@ -86,6 +87,7 @@ def inference(
86
  generator=generator,
87
  ).images
88
 
 
89
  if num_images % 2 == 1:
90
  image = make_image_grid(images, rows=num_images, cols=1)
91
  else:
 
75
 
76
  generator = torch.Generator(device=device).manual_seed(seed)
77
 
78
+ progress(0.3, 'Generating images...')
79
  images = pipe(
80
  prompt,
81
  negative_prompt=negative_prompt,
 
87
  generator=generator,
88
  ).images
89
 
90
+ progress(0.9, f'Done generating {num_images} images')
91
  if num_images % 2 == 1:
92
  image = make_image_grid(images, rows=num_images, cols=1)
93
  else:
requirements.txt CHANGED
@@ -11,3 +11,4 @@ spaces
11
  Pillow
12
  protobuf
13
  peft
 
 
11
  Pillow
12
  protobuf
13
  peft
14
+ compel