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

optional parameter i think

Browse source
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()