fix dtype (#583)

This commit is contained in:
lllyasviel 2023-10-08 15:15:51 -07:00 committed by GitHub
parent d631cfe491
commit 4ff670c7fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions

View File

@ -227,10 +227,10 @@ def patch_model(model, tasks):
for ip_conds, cn_stop, cn_weight in tasks:
if current_step < cn_stop:
ip_k_c = ip_conds[ip_index * 2].to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
ip_v_c = ip_conds[ip_index * 2 + 1].to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
ip_k_uc = ip_unconds[ip_index * 2].to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
ip_v_uc = ip_unconds[ip_index * 2 + 1].to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
ip_k_c = ip_conds[ip_index * 2].to(q)
ip_v_c = ip_conds[ip_index * 2 + 1].to(q)
ip_k_uc = ip_unconds[ip_index * 2].to(q)
ip_v_uc = ip_unconds[ip_index * 2 + 1].to(q)
ip_k = torch.cat([(ip_k_c, ip_k_uc)[i] for i in cond_or_uncond], dim=0)
ip_v = torch.cat([(ip_v_c, ip_v_uc)[i] for i in cond_or_uncond], dim=0)

View File

@ -1 +1 @@
version = '2.1.08'
version = '2.1.09'