File size: 5,969 Bytes
824b515 |
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 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 |
from importlib_resources import files
import os
import functools
import random
import inflect
IE = inflect.engine()
ASSETS_PATH = files("assets")
# ASSETS_PATH = "assets"
@functools.lru_cache(maxsize=None)
def _load_lines(path):
"""
Load lines from a file. First tries to load from `path` directly, and if that doesn't exist, searches the
`ddpo_pytorch/assets` directory for a file named `path`.
"""
if not os.path.exists(path):
newpath = ASSETS_PATH.joinpath(path)
if not os.path.exists(newpath):
raise FileNotFoundError(f"Could not find {path} or assets/{path}")
path = newpath
with open(path, "r") as f:
return [line.strip() for line in f.readlines()]
def hps_v2_all(nouns_file=None, activities_file=None):
return from_file("hps_v2_all.txt")
def hps_custom(nouns_file=None, activities_file=None):
return from_file("hps_custom.txt")
def hps_debug(nouns_file=None, activities_file=None):
return from_file("hps_debug.txt")
def hps_single(nouns_file=None, activities_file=None):
return from_file("hps_single.txt")
def kinetics_4rand(nouns_file=None, activities_file=None):
return from_file("kinetics_4rand.txt")
def kinetics_50rand(nouns_file=None, activities_file=None):
return from_file("kinetics_50rand.txt")
def simple_animals():
return from_file("simple_animals.txt")
def eval_simple_animals():
return from_file("eval_simple_animals.txt")
def eval_hps_v2_all(nouns_file=None, activities_file=None):
return from_file("hps_v2_all_eval.txt")
def chatgpt_custom(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom.txt")
def chatgpt_custom_instruments(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_instruments.txt")
def chatgpt_custom_human(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_human.txt")
def chatgpt_custom_human_activity(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_human_activity.txt")
def chatgpt_custom_animal(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal.txt")
def chatgpt_custom_animal_sport(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_sport.txt")
def chatgpt_custom_animal_sportV2(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_sportV2.txt")
def chatgpt_custom_animal_clothes(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_clothes.txt")
def chatgpt_custom_animal_clothesV2(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_clothesV2.txt")
def chatgpt_custom_animal_clothesV3(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_clothesV3.txt")
def chatgpt_custom_animal_technology(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_technology.txt")
def chatgpt_custom_animal_housework(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_housework.txt")
def chatgpt_custom_animal_action(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_animal_action.txt")
def chatgpt_custom_outdoor(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_outdoor.txt")
def chatgpt_custom_rainy(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_rainy.txt")
def chatgpt_custom_snowy(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_snowy.txt")
def chatgpt_custom_dog(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_dog.txt")
def chatgpt_custom_banana(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_banana.txt")
def chatgpt_custom_forest(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_forest.txt")
def chatgpt_custom_forest_vivid(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_forest_vivid.txt")
def chatgpt_custom_cruel_animal(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_cruel_animal.txt")
def chatgpt_custom_cruel_animal2(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_cruel_animal2.txt")
def chatgpt_custom_bottle_glass(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_bottle_glass.txt")
def chatgpt_custom_book_cup(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_book_cup.txt")
def chatgpt_custom_book_cup_character(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_book_cup_character.txt")
def chatgpt_custom_cute(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_cute.txt")
def chatgpt_custom_ice(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_ice.txt")
def chatgpt_custom_compression(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_compression.txt")
def chatgpt_custom_compression_animals(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_compression_animals.txt")
def chatgpt_custom_actpred(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_actpred.txt")
def chatgpt_custom_actpred2(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_actpred2.txt")
def chatgpt_custom_instruments_unseen(nouns_file=None, activities_file=None):
return from_file("chatgpt_custom_instruments_unseen.txt")
def from_file(path, low=None, high=None, **kwargs):
prompts = _load_lines(path)[low:high]
return random.choice(prompts), {}
def from_str(_str, **kwargs):
return _str, {}
def nouns_activities(nouns_file, activities_file, **kwargs):
nouns = _load_lines(nouns_file)
activities = _load_lines(activities_file)
return f"{IE.a(random.choice(nouns))} {random.choice(activities)}", {} |