Add LoRA support

2023-03-16 21:31:39 -03:00
parent ee164d1821
commit 104293f411
6 changed files with 51 additions and 8 deletions
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -2,7 +2,8 @@ import argparse

 model = None
 tokenizer = None
-model_name = ""
+model_name = "None"
+lora_name = "None"
 soft_prompt_tensor = None
 soft_prompt = False
 is_RWKV = False