dvir-bria commited on
Commit
24591c6
1 Parent(s): 7a35fbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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,