Cognomen commited on
Commit
d162af3
Β·
1 Parent(s): a6ad3d0

remove torchvision pil conversion

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def infer(prompt, negative_prompt, image):
41
  # implement your inference function here
42
  inp = Image.fromarray(image)
43
 
44
- cond_input = F.to_pil_image((conditioning_image_transforms(inp))[0].to("cuda"))
45
  #cond_input = T.ToPILImage(cond_input)
46
 
47
  output = pipe(
 
41
  # implement your inference function here
42
  inp = Image.fromarray(image)
43
 
44
+ cond_input = conditioning_image_transforms(inp)[0]
45
  #cond_input = T.ToPILImage(cond_input)
46
 
47
  output = pipe(