array-in-a-matrix
/
text-gen-bot
1
0
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

optional parameter i think

This commit is contained in:
array-in-a-matrix 2022-08-22 00:27:32 -04:00
parent ac80edae2d
commit 54b451908a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
textgen.py
View File

@ -13,7 +13,7 @@ file_name = json_object['file']
# ? generate message using trained model
def generate_message(prompt):
def generate_message(*prompt):
ai = aitextgen(model_folder="trained_model",
tokenizer_file="aitextgen.tokenizer.json", prompt=prompt)
ai.generate()