Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,7 +106,7 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
|
|
| 106 |
@spaces.GPU
|
| 107 |
def generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed):
|
| 108 |
pipe.to('cuda')
|
| 109 |
-
generator = torch.manual_seed(seed)
|
| 110 |
images = pipe(
|
| 111 |
prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|
| 112 |
generator=generator,
|
|
|
|
| 106 |
@spaces.GPU
|
| 107 |
def generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed):
|
| 108 |
pipe.to('cuda')
|
| 109 |
+
generator = torch.Generator("cuda").manual_seed(seed)
|
| 110 |
images = pipe(
|
| 111 |
prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|
| 112 |
generator=generator,
|