Kannon commited on
Commit
e2c4aa0
·
1 Parent(s): ac7f4d9

Delete constants.py

Browse files
Files changed (1) hide show
  1. constants.py +0 -109
constants.py DELETED
@@ -1,109 +0,0 @@
1
- # Constants
2
- # Also try: 'Qiliang/bart-large-cnn-samsum-ElectrifAi_v10'
3
- DEFAULT_SUMMARIZATION_MODEL = "Qiliang/bart-large-cnn-samsum-ChatGPT_v3"
4
- # Also try: 'joeddav/distilbert-base-uncased-go-emotions-student'
5
- DEFAULT_CLASSIFICATION_MODEL = "nateraw/bert-base-uncased-emotion"
6
- # Also try: 'Salesforce/blip-image-captioning-base'
7
- DEFAULT_CAPTIONING_MODEL = "Salesforce/blip-image-captioning-large"
8
- DEFAULT_KEYPHRASE_MODEL = "ml6team/keyphrase-extraction-distilbert-inspec"
9
- DEFAULT_PROMPT_MODEL = "FredZhang7/anime-anything-promptgen-v2"
10
- DEFAULT_SD_MODEL = "ckpt/anything-v4.5-vae-swapped"
11
- DEFAULT_EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
12
- DEFAULT_REMOTE_SD_HOST = "127.0.0.1"
13
- DEFAULT_REMOTE_SD_PORT = 7860
14
- SILERO_SAMPLES_PATH = "tts_samples"
15
- SILERO_SAMPLE_TEXT = "The quick brown fox jumps over the lazy dog"
16
- # ALL_MODULES = ['caption', 'summarize', 'classify', 'keywords', 'prompt', 'sd']
17
- DEFAULT_SUMMARIZE_PARAMS = {
18
- "temperature": 1.0,
19
- "repetition_penalty": 1.0,
20
- "max_length": 500,
21
- "min_length": 200,
22
- "length_penalty": 1.5,
23
- "bad_words": [
24
- "\n",
25
- '"',
26
- "*",
27
- "[",
28
- "]",
29
- "{",
30
- "}",
31
- ":",
32
- "(",
33
- ")",
34
- "<",
35
- ">",
36
- "Â",
37
- "The text ends",
38
- "The story ends",
39
- "The text is",
40
- "The story is",
41
- ],
42
- }
43
-
44
- PROMPT_PREFIX = "best quality, absurdres, "
45
- NEGATIVE_PROMPT = """lowres, bad anatomy, error body, error hair, error arm,
46
- error hands, bad hands, error fingers, bad fingers, missing fingers
47
- error legs, bad legs, multiple legs, missing legs, error lighting,
48
- error shadow, error reflection, text, error, extra digit, fewer digits,
49
- cropped, worst quality, low quality, normal quality, jpeg artifacts,
50
- signature, watermark, username, blurry"""
51
-
52
-
53
- # list of key phrases to be looking for in text (unused for now)
54
- INDICATOR_LIST = [
55
- "female",
56
- "girl",
57
- "male",
58
- "boy",
59
- "woman",
60
- "man",
61
- "hair",
62
- "eyes",
63
- "skin",
64
- "wears",
65
- "appearance",
66
- "costume",
67
- "clothes",
68
- "body",
69
- "tall",
70
- "short",
71
- "chubby",
72
- "thin",
73
- "expression",
74
- "angry",
75
- "sad",
76
- "blush",
77
- "smile",
78
- "happy",
79
- "depressed",
80
- "long",
81
- "cold",
82
- "breasts",
83
- "chest",
84
- "tail",
85
- "ears",
86
- "fur",
87
- "race",
88
- "species",
89
- "wearing",
90
- "shoes",
91
- "boots",
92
- "shirt",
93
- "panties",
94
- "bra",
95
- "skirt",
96
- "dress",
97
- "kimono",
98
- "wings",
99
- "horns",
100
- "pants",
101
- "shorts",
102
- "leggins",
103
- "sandals",
104
- "hat",
105
- "glasses",
106
- "sweater",
107
- "hoodie",
108
- "sweatshirt",
109
- ]