| from lyraBelle import LyraBelle | |
| data_type = "fp16" | |
| prompts = "今天天气大概 25度，有点小雨，吹着风，我想去户外散步，应该穿什么样的衣服裤子鞋子搭配。" | |
| model_dir = "./model" | |
| model_name = "1-gpu-fp16.h5" | |
| max_output_length = 512 | |
| model = LyraBelle(model_dir, model_name, data_type, 0) | |
| output_texts = model.generate(prompts, output_length=max_output_length,top_k=30, top_p=0.85, temperature=0.35, repetition_penalty=1.2, do_sample=True) | |
| print(output_texts) |