Spaces:
Build error
Build error
File size: 5,697 Bytes
60675e2 1f1a749 60675e2 99044e4 60675e2 1f1a749 60675e2 1f1a749 99044e4 7e1c4ca 60675e2 1f1a749 7e1c4ca 99044e4 60675e2 99044e4 60675e2 1f1a749 60675e2 cab48de 60675e2 99044e4 60675e2 99044e4 60675e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
from llama_cpp import Llama
from huggingface_hub import hf_hub_download
import gradio as gr
import random
repo_id = "TheBloke/Mistral-7B-Instruct-v0.1-GGUF"
filename = "mistral-7b-instruct-v0.1.Q4_K_M.gguf"
model_file_path = hf_hub_download(repo_id=repo_id, filename=filename)
def generate_characters(generate_backstories=False):
class_list = ["Barbarian", "Bard", "Cleric", "Druid", "Fighter",
"Monk", "Paladin", "Ranger", "Rogue", "Sorcerer", "Warlock", "Wizard"]
sub_class_list = ["Barbarian - Beserker", "Barbarian - Wildheart", "Barbarian - Wild Magic", "Bard - College of Lore", "Bard - College of Valour", "Bard - College of Swords", "Cleric - Life Domain", "Cleric - Light Domain", "Cleric - Trickery Domain", "Cleric - Knowledge Domain", "Cleric - Nature Domain", "Cleric - Tempest Domain", "Cleric - War Domain", "Druid - Circle of the moon", "Druid - Circle of the Land", "Druid - Circle of the Spores", "Fighter - Battle Master", "Fighter - Eldritch Knight", "Fighter - Champion", "Monk - Way of the open hand", "Monk - Way of shadow", "Monk - Way of the four elements",
"Paladin - Oath of Devotion", "Paladin - Oath of the ancients", "Paladin - Oath of Vengence", "Paladin - Oathbreaker", "Ranger - Beat Master", "Ranger - Hunter", "Ranger - Gloom Stalker", "Rogue - Thief", "Rogue - Arcane Trickster", "Rogue - Assassin", "Sorcerer - Draconic Bloodline", "Sorcerer - Wild Magic", "Sorcerer - Storm Sorcery", "Warlock - The Fiend", "Warlock - the great old one", "Warlock - Archfey", "Wizard - Abjuration", "Wizard - Conjuration", "Wizard - Divination", "Wizard - Enchantment", "Wizard - Evocation", "Wizard - Necromancer", "Wizard - Illusion", "Wizard - Transmutation"]
race_list = ["Dragonborn", "Drow", "Dwarf", "Elf", "Githyanki",
"Gnome", "Half-Elf", "Half-Orc", "Halfling", "Human", "Tiefling"]
sub_race_list = ["Black Dragonborn", "Blue Dragonborn", "Brass Dragonborn", "Bronze Dragonborn", "Copper Dragonborn", "Gold Dragonborn", "Green Dragonborn", "Red Dragonborn", "Silver Dragonborn", "White Dragonborn", "Lolth-Sworn Drow", "Seldarine Drow", "Gold Dwarf", "Shield Dwarf",
"Duegar (Dwarf)", "High Elf - Elf", "Wood Elf - Elf", "Githyanki", "Deep Gnome", "Forest Gnome", "Rock Gnome", "High Half-Elf", "Wood Half-Elf", "Drow Half-Elf", "Half-Orc", "Lightfoot Halfling", "Strongheart Halfling", "Human", "Asmodeus Tiefling", "Mephistopheles Tiefling", "Zariel Tiefling"]
alignment_list = ["Chaotic Good", "Lawful Good", "Neutral Good",
"Chaotic Evil", "Lawful Evil", "Neutral Evil", "True Chaotic", "True Neutral"]
background_list = ["Acolyte", "Charlatan", "Criminal", "Entertainer",
"Folk Hero", "Guild Artisan", "Noble", "Hermit", "Outlander", "Sage", "Soldier"]
def generate_character():
character = {
'Class': random.choice(class_list),
'Sub Class': random.choice(sub_class_list),
'Race': random.choice(race_list),
'Sub Race': random.choice(sub_race_list),
'Alignment': random.choice(alignment_list),
'Background': random.choice(background_list),
}
return character
def generate_multiclass_character():
character = generate_character()
second_class = random.choice(
[cls for cls in class_list if cls != character['Class']])
levels_class1 = random.randint(1, 11)
levels_class2 = 12 - levels_class1
character['Class'] += f' ({levels_class1} levels), {second_class} ({levels_class2} levels)'
return character
characters = [generate_character() for _ in range(2)]
characters.append(generate_multiclass_character())
def create_character_output(character, generate_backstories=False):
output = ""
for key, value in character.items():
output += f"- **{key}:** {value}\n"
output += "\n"
if generate_backstories:
backstory = generate_backstory(character)
# Remove everything up to and including `[/INST]\n`
output += backstory[backstory.find("[/INST]\n") + 8:]
return output
output = ""
for idx, char in enumerate(characters, 1):
output += f"### Character {idx}:\n"
output += create_character_output(char, generate_backstories)
return output
def generate_text_from_prompt(user_prompt,
max_tokens=4096,
temperature=1.0,
top_p=0.8,
echo=True,
stop=["<s>", "[/INST]", "Q"]):
mistral_model = Llama(model_path=model_file_path,
n_ctx=4096,
n_gpu_layers=24)
# Define the parameters
model_output = mistral_model(
f"<s>[INST] {user_prompt} [/INST]",
max_tokens=max_tokens,
temperature=temperature,
top_p=top_p,
echo=echo,
stop=stop
)
return model_output
def generate_backstory(character):
return generate_text_from_prompt(
f"Please provide a name and creative, well-written backstory "
f" for the following D&D Character: \n```{character}```\n"
)
if __name__ == "__main__":
iface = gr.Interface(fn=generate_characters,
title="D&D Character Generator",
inputs=[gr.inputs.Checkbox(
label="Generate Backstories?")],
description="Generates 3 random D&D characters.",
outputs="markdown")
iface.launch()
|