From 48485fe92f01a9de71174286017588788b998796 Mon Sep 17 00:00:00 2001 From: user1 Date: Thu, 11 May 2023 13:54:32 -0700 Subject: [PATCH] Removed last bits of dtype and and device hardwiring from controlnet section --- invokeai/app/invocations/latent.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py index 74952e7896..1d1eb1963a 100644 --- a/invokeai/app/invocations/latent.py +++ b/invokeai/app/invocations/latent.py @@ -300,15 +300,9 @@ class TextToLatentsInvocation(BaseInvocation): control_weights.append(control_info.control_weight) # handle control models - # FIXME: change this to dropdown menu? - # FIXME: generalize so don't have to hardcode torch_dtype and device + # FIXME: change this to dropdown menu control_model = ControlNetModel.from_pretrained(control_info.control_model, - #torch_dtype=model.unet.dtype).to(model.device) - #torch.dtype=model.unet.dtype).to("cuda") - # torch.dtype = model.unet.dtype).to("cuda") - torch_dtype=torch.float16).to("cuda") - # torch_dtype = torch.float16).to(model.device) - # model.dtype).to(model.device) + torch_dtype=model.unet.dtype).to(model.device) control_models.append(control_model) # handle control images