Spaces:
Runtime error
Runtime error
convert to fp32
Browse files
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,
|