Don't show 'None' in the LoRA list

This commit is contained in:
oobabooga 2023-04-17 13:52:23 -03:00
parent 19e3a59997
commit c3dc348d1c
2 changed files with 6 additions and 5 deletions

View file

@ -73,7 +73,7 @@ def get_available_softprompts():
def get_available_loras():
return ['None'] + sorted([item.name for item in list(Path(shared.args.lora_dir).glob('*')) if not item.name.endswith(('.txt', '-np', '.pt', '.json'))], key=str.lower)
return sorted([item.name for item in list(Path(shared.args.lora_dir).glob('*')) if not item.name.endswith(('.txt', '-np', '.pt', '.json'))], key=str.lower)
def load_model_wrapper(selected_model):