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,
|