Add LoRA support
This commit is contained in:
parent
ee164d1821
commit
104293f411
6 changed files with 51 additions and 8 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue