Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed last bits of dtype and and device hardwiring from controlnet section

Browse Source
This commit is contained in:
user1 2023-05-11 13:54:32 -07:00 committed by Kent Keirsey
parent 07726af703
commit 48485fe92f
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/app/invocations/latent.py
View File

@ -300,15 +300,9 @@ class TextToLatentsInvocation(BaseInvocation):
control_weights.append(control_info.control_weight) control_weights.append(control_info.control_weight)
# handle control models # handle control models
# FIXME: change this to dropdown menu? # FIXME: change this to dropdown menu
# FIXME: generalize so don't have to hardcode torch_dtype and device
control_model = ControlNetModel.from_pretrained(control_info.control_model, control_model = ControlNetModel.from_pretrained(control_info.control_model,
#torch_dtype=model.unet.dtype).to(model.device) 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)
control_models.append(control_model) control_models.append(control_model)
# handle control images # handle control images