LTT commited on
Commit
6f2632d
·
verified ·
1 Parent(s): f679c8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -208,7 +208,7 @@ def generate_multi_view_images(prompt, seed):
208
  # torch.cuda.empty_cache()
209
  # generator = torch.manual_seed(seed)
210
  generator = torch.Generator().manual_seed(seed)
211
- # with torch.no_grad():
212
  # images = flux_pipe(
213
  # prompt=prompt,
214
  # num_inference_steps=10,
@@ -220,10 +220,10 @@ def generate_multi_view_images(prompt, seed):
220
  # generator=generator,
221
  # good_vae=good_vae,
222
  # ).images
223
- for img in flux_pipe.flux_pipe_call_that_returns_an_iterable_of_images(
224
  prompt=prompt,
225
  guidance_scale=3.5,
226
- num_inference_steps=20,
227
  width=resolution * 4,
228
  height=resolution * 2,
229
  generator=generator,
 
208
  # torch.cuda.empty_cache()
209
  # generator = torch.manual_seed(seed)
210
  generator = torch.Generator().manual_seed(seed)
211
+ with torch.no_grad():
212
  # images = flux_pipe(
213
  # prompt=prompt,
214
  # num_inference_steps=10,
 
220
  # generator=generator,
221
  # good_vae=good_vae,
222
  # ).images
223
+ for img in flux_pipe.flux_pipe_call_that_returns_an_iterable_of_images(
224
  prompt=prompt,
225
  guidance_scale=3.5,
226
+ num_inference_steps=15,
227
  width=resolution * 4,
228
  height=resolution * 2,
229
  generator=generator,