Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 1, 2023

Commit

e8933b1

1 Parent(s): bca8e13

use fp32 for cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,11 +19,11 @@ conditioning_image_transforms = T.Compose(
     ]
 )
-cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float16, from_flax=True)
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
         "./models/wd-1-5-b2",
         controlnet=cnet,
-        torch_dtype=torch.float16,
         )
 generator = torch.manual_seed(0)

     ]
 )
+cnet = ControlNetModel.from_pretrained("./models/catcon-controlnet-wd", torch_dtype=torch.float32, from_flax=True)
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
         "./models/wd-1-5-b2",
         controlnet=cnet,
+        torch_dtype=torch.float32,
         )
 generator = torch.manual_seed(0)