Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Grammar

Browse Source
This commit is contained in:
oobabooga 2023-01-09 19:07:47 -03:00
parent 7028116bf2
commit 3aefcfd963
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -91,7 +91,7 @@ Optionally, you can use the following command-line flags:
to the same text box as the input. to the same text box as the input.
--chat Launch the webui in chat mode. --chat Launch the webui in chat mode.
--cpu Use the CPU to generate text. --cpu Use the CPU to generate text.
--listen Makes the webui reachable from your local network. --listen Make the webui reachable from your local network.
``` ```
## Presets ## Presets

2
server.py
View File

@ -17,7 +17,7 @@ parser.add_argument('--model', type=str, help='Name of the model to load by defa
parser.add_argument('--notebook', action='store_true', help='Launch the webui in notebook mode, where the output is written to the same text box as the input.') parser.add_argument('--notebook', action='store_true', help='Launch the webui in notebook mode, where the output is written to the same text box as the input.')
parser.add_argument('--chat', action='store_true', help='Launch the webui in chat mode.') parser.add_argument('--chat', action='store_true', help='Launch the webui in chat mode.')
parser.add_argument('--cpu', action='store_true', help='Use the CPU to generate text.') parser.add_argument('--cpu', action='store_true', help='Use the CPU to generate text.')
parser.add_argument('--listen', action='store_true', help='Makes the webui reachable from your local network.') parser.add_argument('--listen', action='store_true', help='Make the webui reachable from your local network.')
args = parser.parse_args() args = parser.parse_args()
loaded_preset = None loaded_preset = None
available_models = sorted(set(map(lambda x : str(x.name).replace('.pt', ''), list(Path('models/').glob('*'))+list(Path('torch-dumps/').glob('*'))))) available_models = sorted(set(map(lambda x : str(x.name).replace('.pt', ''), list(Path('models/').glob('*'))+list(Path('torch-dumps/').glob('*')))))