Update app.py

app.py

@@ -38,7 +38,8 @@ def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resoluti
         # detected_map = apply_canny(img, low_threshold, high_threshold)
         # detected_map = HWC3(detected_map)
 
-        control = torch.from_numpy(img.copy()).float().cuda() / 255.0
+        #control = torch.from_numpy(img.copy()).float().cuda() / 255.0
+        control = torch.from_numpy(img.copy()).float() / 255.0
         control = torch.stack([control for _ in range(num_samples)], dim=0)
         control = einops.rearrange(control, 'b h w c -> b c h w').clone()
         control = control.to(memory_format=torch.contiguous_format).float()