mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Further refactor

Browse source
This commit is contained in:
oobabooga 2023-02-23 13:28:30 -03:00
parent ce7feb3641
commit 1dacd34165
8 changed files with 185 additions and 173 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/shared.py
View file

@ -6,6 +6,7 @@ model_name = ""
soft_prompt_tensor = None
soft_prompt = False
stop_everything = False
settings = {}
parser = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog,max_help_position=54))
parser.add_argument('--model', type=str, help='Name of the model to load by default.')