kth_test / template.config
krpg's picture
Upload template.config with huggingface_hub
7b929f9 verified
raw
history blame
2.76 kB
{
"model_name": "kth_test",
"base_model": "unsloth/Qwen2-0.5B",
"train_dataset": "keith_sample",
"test_dataset": "keith_sample",
"rank": 32,
"alpha": 32,
"epoch": 1,
"learning_rate": 5e-05,
"learning_rate_scheduler": "linear",
"template": {
"name": "Heroic Deeds",
"version": "0.0.1",
"template_description": "The template for creating a model that can create heroic deeds",
"inputs": [
{
"variable_name": "author",
"type": "str",
"description": "The cannonical author of the heroic deed text"
},
{
"variable_name": "character",
"type": "str",
"description": "The character the heroic deed text is about"
},
{
"variable_name": "emotions",
"type": "List[str]",
"description": "A list of one-word emotions explaining the general emotion the text is trying to convey"
},
{
"variable_name": "genre",
"type": "str",
"description": "The literary genre of the work"
},
{
"variable_name": "genre_description",
"type": "str",
"description": "The description of the genre, providing guidelines as to how to use it."
},
{
"variable_name": "heroic_deed",
"type": "str",
"description": "A heroic deed is a notable achievement for a given character, that has profound effects on the surrounding community and their social standing"
}
],
"output": {
"Author": "str",
"Genre": "str",
"Title": "str",
"Text": "str"
},
"prompt": "You are a master word-smith, primed to create exciting tales from medieval europe. You will be given an author ({{author.description}}), a main character ({{character.description}}), a heroic deed ({{heroic_deed.description}}), a few emotions ({{emotions.description}}), a literary genre, as well as an explenation as to what that literary genre entails ({{genre_description}}) and you are meant to write a text based on that. \n You will ALWAYS reply in the following valid JSON: { 'Author': 'author', 'Title': 'title', 'Text': 'the generated text' }. \n Please write a text in the style of {{genre}} ({{genre_description}}), from the perspective of {{author}} regarding {{character}} about their {{heroic_deed}}. The text should be {{emotions}} and you should generate a suitable title for it as well. REMEMBER TO OUTPUT THE VALID JSON DESCRIBED ABOVE."
}
}