diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-02-19 09:41:29 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-02-19 09:41:29 +0000 |
| commit | dfb3b8f398239876cdfc03657680e50c76a1fed8 (patch) | |
| tree | 30b5b4e5a179e9d88e75ac0abdcf0090e1c74397 /modules/hypernetworks/hypernetwork.py | |
| parent | edb10092de516dda5271130ed53628387780a859 (diff) | |
| parent | 43137773227344dd79b56c227dbc9fb54ac7c337 (diff) | |
| download | stable-diffusion-webui-gfx803-dfb3b8f398239876cdfc03657680e50c76a1fed8.tar.gz stable-diffusion-webui-gfx803-dfb3b8f398239876cdfc03657680e50c76a1fed8.tar.bz2 stable-diffusion-webui-gfx803-dfb3b8f398239876cdfc03657680e50c76a1fed8.zip | |
Merge branch 'master' into weighted-learning
Diffstat (limited to 'modules/hypernetworks/hypernetwork.py')
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index f4fb69e0..f6ef42d5 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -380,8 +380,8 @@ def apply_single_hypernetwork(hypernetwork, context_k, context_v, layer=None): layer.hyper_k = hypernetwork_layers[0]
layer.hyper_v = hypernetwork_layers[1]
- context_k = hypernetwork_layers[0](context_k)
- context_v = hypernetwork_layers[1](context_v)
+ context_k = devices.cond_cast_unet(hypernetwork_layers[0](devices.cond_cast_float(context_k)))
+ context_v = devices.cond_cast_unet(hypernetwork_layers[1](devices.cond_cast_float(context_v)))
return context_k, context_v
|