lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use computation_dtype instead of storage_dtype

Browse Source
This commit is contained in:
wcole3
2024-09-02 12:23:17 -04:00
parent 350018eff5
commit dee7a3c287
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions-builtin/sd_forge_controlllite/lib_controllllite/lib_controllllite.py
View File

@@ -267,7 +267,7 @@ class LLLiteLoader:
model_lllite = model.clone()
patch = load_control_net_lllite_patch(state_dict, cond_image, strength, steps, start_percent, end_percent,
model_lllite.model.storage_dtype)
model.model.diffusion_model.computation_dtype)
if patch is not None:
model_lllite.set_model_attn1_patch(patch)
model_lllite.set_model_attn2_patch(patch)