This commit is contained in:
wassname
2023-04-15 06:37:43 +00:00
parent c520ad64ba
commit e6338b5b26
3 changed files with 15 additions and 126 deletions
+4
View File
@@ -27,6 +27,10 @@ tokenizer = LlamaTokenizer.from_pretrained(args.model)
outs = test_conversation(model, tokenizer)
print(outs)
prompts_path = Path(output_path) / 'test_prompts2.txt'
prompts_path.open('w').write(o)
print(prompts_path)
# from transformers import pipeline, Conversation, ConversationalPipeline
# # https://huggingface.co/tasks/conversational
# prompts = [