diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2023-02-06 18:10:55 -0500 |
|---|---|---|
| committer | brkirch <brkirch@users.noreply.github.com> | 2023-02-06 18:10:55 -0500 |
| commit | 4738486d8f528a98a525970ac06a109431fd7344 (patch) | |
| tree | 2d01a386f2436a064b903d785dadeaab0279fdeb /modules | |
| parent | ea9bd9fc7409109adcd61b897abc2c8881161256 (diff) | |
Support for hypernetworks with --upcast-sampling
Diffstat (limited to 'modules')
| -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 825a93b2..a15bae18 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
|