You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
9 lines
390 B
9 lines
390 B
from aitextgen import aitextgen
|
|
|
|
if __name__ == '__main__':
|
|
# With your trained model, you can reload the model at any time by
|
|
# providing the folder containing the pytorch_model.bin model weights + the config, and providing the tokenizer.
|
|
ai2 = aitextgen(model_folder="trained_model",
|
|
tokenizer_file="aitextgen.tokenizer.json")
|
|
|
|
ai2.generate(10)
|