diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-21 13:15:53 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-21 13:15:53 +0000 |
| commit | 855b9e3d1c5a1bd8c2d815d38a38bc7c410be5a8 (patch) | |
| tree | 6c978dd9650c65575ea26d3e86480fc383923076 /extensions-builtin/Lora/extra_networks_lora.py | |
| parent | cbfb4632585415dc914aff8c44869d792fd64c24 (diff) | |
| download | stable-diffusion-webui-gfx803-855b9e3d1c5a1bd8c2d815d38a38bc7c410be5a8.tar.gz stable-diffusion-webui-gfx803-855b9e3d1c5a1bd8c2d815d38a38bc7c410be5a8.tar.bz2 stable-diffusion-webui-gfx803-855b9e3d1c5a1bd8c2d815d38a38bc7c410be5a8.zip | |
Lora support!
update readme to reflect some recent changes
Diffstat (limited to 'extensions-builtin/Lora/extra_networks_lora.py')
| -rw-r--r-- | extensions-builtin/Lora/extra_networks_lora.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/extra_networks_lora.py b/extensions-builtin/Lora/extra_networks_lora.py new file mode 100644 index 00000000..8f2e753e --- /dev/null +++ b/extensions-builtin/Lora/extra_networks_lora.py @@ -0,0 +1,20 @@ +from modules import extra_networks
+import lora
+
+class ExtraNetworkLora(extra_networks.ExtraNetwork):
+ def __init__(self):
+ super().__init__('lora')
+
+ def activate(self, p, params_list):
+ names = []
+ multipliers = []
+ for params in params_list:
+ assert len(params.items) > 0
+
+ names.append(params.items[0])
+ multipliers.append(float(params.items[1]) if len(params.items) > 1 else 1.0)
+
+ lora.load_loras(names, multipliers)
+
+ def deactivate(self, p):
+ pass
|