Add lora directory helper

pull/201/head
ExponentialML 2023-07-12 12:38:07 -07:00 committed by GitHub
parent 2dad959835
commit 4ebcbd1135
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -11,7 +11,7 @@ import os
import json
from modules import images, script_callbacks
from modules.shared import opts, cmd_opts, state, cmd_opts, sd_model
from modules.shared import opts, state, cmd_opts, sd_model
from modules.sd_models import read_state_dict
from stable_lora.stable_utils.lora_processor import StableLoraProcessor
from t2v_helpers.extensions_utils import Text2VideoExtension
@ -73,6 +73,7 @@ class StableLoraScript(Text2VideoExtension, StableLoraProcessor):
To train a Stable LoRA file, use the finetune repository by clicking here.
</a>"""
)
gr.HTML(f"<span> Place your LoRA files in {cmd_opts.lora_dir}")
lora_files_selection = gr.Dropdown(
label="Available Models",
elem_id=GradioInputsIds.lora_file_selection,