From 728e7c79fd668400a742c234a54aa3d214c5b9dd Mon Sep 17 00:00:00 2001
From: patriceac <48073125+patriceac@users.noreply.github.com>
Date: Tue, 28 Mar 2023 03:07:50 -0700
Subject: [PATCH] Add support for LoRA models to dnd.js (#1054)

* Add support for LoRA to dnd.js

Adds support for LoRA to dnd.js (task restoration, use settings, etc.).

* Correct extensions for LoRA

---------

Co-authored-by: cmdr2 <shashank.shekhar.global@gmail.com>
---
 ui/media/js/dnd.js | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/ui/media/js/dnd.js b/ui/media/js/dnd.js
index f1f00911..3c02fa1a 100644
--- a/ui/media/js/dnd.js
+++ b/ui/media/js/dnd.js
@@ -230,6 +230,20 @@ const TASK_MAPPING = {
         readUI: () => vaeModelField.value,
         parse: (val) => val
     },
+    use_lora_model: { name: 'LoRA model',
+        setUI: (use_lora_model) => {
+            const oldVal = loraModelField.value
+            use_lora_model = (use_lora_model === undefined || use_lora_model === null || use_lora_model === 'None' ? '' : use_lora_model)
+
+            if (use_lora_model !== '') {
+                use_lora_model = getModelPath(use_lora_model, ['.ckpt', '.safetensors'])
+                use_lora_model = use_lora_model !== '' ? use_lora_model : oldVal
+            }
+            loraModelField.value = use_lora_model
+        },
+        readUI: () => loraModelField.value,
+        parse: (val) => val
+    },
     use_hypernetwork_model: { name: 'Hypernetwork model',
         setUI: (use_hypernetwork_model) => {
             const oldVal = hypernetworkModelField.value