Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -94,7 +94,9 @@ def get_canny_filter(image):
|
|
94 |
|
95 |
@spaces.GPU
|
96 |
def generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed):
|
97 |
-
generator = torch.Generator("cuda").manual_seed(seed)
|
|
|
|
|
98 |
images = pipe(
|
99 |
prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|
100 |
generator=generator,
|
|
|
94 |
|
95 |
@spaces.GPU
|
96 |
def generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed):
|
97 |
+
generator = torch.Generator("cuda").manual_seed(seed)
|
98 |
+
print(pipe.unet.dtype)
|
99 |
+
print(pipe.controlnet.dtype)
|
100 |
images = pipe(
|
101 |
prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|
102 |
generator=generator,
|