Cognomen commited on
Commit
8c5b762
Β·
1 Parent(s): bb3c6b8

convert to fp32

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ def infer(prompt, negative_prompt, image):
42
  inp = Image.fromarray(image)
43
 
44
  cond_input = conditioning_image_transforms(inp)
45
- cond_input = T.ToPILImage()(cond_input.to("cpu"))
46
 
47
  output = pipe(
48
  prompt,
 
42
  inp = Image.fromarray(image)
43
 
44
  cond_input = conditioning_image_transforms(inp)
45
+ cond_input = T.ToPILImage()(cond_input.to("cpu", dtype=torch.float32))
46
 
47
  output = pipe(
48
  prompt,