Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ pipe.enable_model_cpu_offload()
|
|
| 60 |
pipe.enable_xformers_memory_efficient_attention()
|
| 61 |
|
| 62 |
# Generator seed,
|
| 63 |
-
generator = torch.manual_seed(
|
| 64 |
|
| 65 |
def get_canny_filter(image):
|
| 66 |
if not isinstance(image, np.ndarray):
|
|
@@ -78,7 +78,6 @@ def generate_images(image, prompt):
|
|
| 78 |
output = pipe(
|
| 79 |
prompt,
|
| 80 |
canny_image,
|
| 81 |
-
seed=42,
|
| 82 |
generator=generator,
|
| 83 |
num_images_per_prompt=1,
|
| 84 |
num_inference_steps=20,
|
|
|
|
| 60 |
pipe.enable_xformers_memory_efficient_attention()
|
| 61 |
|
| 62 |
# Generator seed,
|
| 63 |
+
generator = torch.manual_seed(42)
|
| 64 |
|
| 65 |
def get_canny_filter(image):
|
| 66 |
if not isinstance(image, np.ndarray):
|
|
|
|
| 78 |
output = pipe(
|
| 79 |
prompt,
|
| 80 |
canny_image,
|
|
|
|
| 81 |
generator=generator,
|
| 82 |
num_images_per_prompt=1,
|
| 83 |
num_inference_steps=20,
|