Update app.py
Browse files
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 |
-
|
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 |
-
|
224 |
prompt=prompt,
|
225 |
guidance_scale=3.5,
|
226 |
-
num_inference_steps=
|
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,
|