Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -148,26 +148,26 @@ def generate_image_for_gradio(
|
|
| 148 |
generator = torch.Generator(device=DEVICE).manual_seed(seed)
|
| 149 |
|
| 150 |
|
| 151 |
-
# with torch.no_grad():
|
| 152 |
-
# generated_images = pipeline(
|
| 153 |
-
# prompt,
|
| 154 |
-
# image=input_image_for_pipeline,
|
| 155 |
-
# num_inference_steps=num_inference_steps,
|
| 156 |
-
# guidance_scale=guidance_scale,
|
| 157 |
-
# generator=generator,
|
| 158 |
-
# ).images
|
| 159 |
-
|
| 160 |
with torch.no_grad():
|
| 161 |
generated_images = pipeline(
|
| 162 |
prompt,
|
| 163 |
-
negative_prompt,
|
| 164 |
image=input_image_for_pipeline,
|
| 165 |
num_inference_steps=25,
|
| 166 |
-
|
| 167 |
-
strength = 0.85,
|
| 168 |
generator=generator,
|
| 169 |
).images
|
| 170 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 171 |
print(f"Image generation complete (seed: {seed}).")
|
| 172 |
return generated_images[0]
|
| 173 |
|
|
|
|
| 148 |
generator = torch.Generator(device=DEVICE).manual_seed(seed)
|
| 149 |
|
| 150 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 151 |
with torch.no_grad():
|
| 152 |
generated_images = pipeline(
|
| 153 |
prompt,
|
|
|
|
| 154 |
image=input_image_for_pipeline,
|
| 155 |
num_inference_steps=25,
|
| 156 |
+
guidance_scale=8.0,
|
|
|
|
| 157 |
generator=generator,
|
| 158 |
).images
|
| 159 |
|
| 160 |
+
# with torch.no_grad():
|
| 161 |
+
# generated_images = pipeline(
|
| 162 |
+
# prompt,
|
| 163 |
+
# negative_prompt,
|
| 164 |
+
# image=input_image_for_pipeline,
|
| 165 |
+
# num_inference_steps=25,
|
| 166 |
+
# # guidance_scale=8.0,
|
| 167 |
+
# strength = 0.85,
|
| 168 |
+
# generator=generator,
|
| 169 |
+
# ).images
|
| 170 |
+
|
| 171 |
print(f"Image generation complete (seed: {seed}).")
|
| 172 |
return generated_images[0]
|
| 173 |
|