AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

use new shared.args.lora_dir

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin 2023-03-27 20:04:16 -07:00
parent 31f04dc615
commit ec6224f556
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/training.py
View File

@ -92,8 +92,7 @@ def do_train(lora_name: str, micro_batch_size: int, batch_size: int, epochs: int
# == Input validation / processing ==
yield "Prepping..."
# TODO: --lora-dir PR once pulled will need to be applied here
lora_name = f"loras/{clean_path(None, lora_name)}"
lora_name = f"{shared.args.lora_dir}/{clean_path(None, lora_name)}"
if dataset is None:
return "**Missing dataset choice input, cannot continue.**"
if format is None: