Spaces:
Running
on
Zero
Running
on
Zero
Upload 3 files
Browse files- genimage.py +1 -1
- llmdolphin.py +1 -0
- tagger.py +12 -6
genimage.py
CHANGED
|
@@ -45,7 +45,7 @@ def generate_image(prompt, neg_prompt):
|
|
| 45 |
try:
|
| 46 |
images = pipe(
|
| 47 |
prompt=prompt,
|
| 48 |
-
prompt_2="
|
| 49 |
negative_prompt=neg_prompt,
|
| 50 |
negative_prompt_2="lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
|
| 51 |
width=1024,
|
|
|
|
| 45 |
try:
|
| 46 |
images = pipe(
|
| 47 |
prompt=prompt,
|
| 48 |
+
prompt_2="masterpiece, best quality, very aesthetic, absurdres",
|
| 49 |
negative_prompt=neg_prompt,
|
| 50 |
negative_prompt_2="lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
|
| 51 |
width=1024,
|
llmdolphin.py
CHANGED
|
@@ -16,6 +16,7 @@ llm_models = {
|
|
| 16 |
"suzume-llama-3-8B-japanese.Q4_K_M.gguf": ["PrunaAI/lightblue-suzume-llama-3-8B-japanese-GGUF-smashed", MessagesFormatterType.LLAMA_3],
|
| 17 |
"suzume-llama-3-8B-multilingual-orpo-borda-top25.Q4_K_M.gguf": ["RichardErkhov/lightblue_-_suzume-llama-3-8B-multilingual-orpo-borda-top25-gguf", MessagesFormatterType.LLAMA_3],
|
| 18 |
"Bungo-L3-8B.Q5_K_M.gguf": ["backyardai/Bungo-L3-8B-GGUF", MessagesFormatterType.LLAMA_3],
|
|
|
|
| 19 |
"Mistral-Nemo-Instruct-2407-Q4_K_M.gguf": ["bartowski/Mistral-Nemo-Instruct-2407-GGUF", MessagesFormatterType.MISTRAL],
|
| 20 |
"ghost-8b-beta.q5_k.gguf": ["ZeroWw/ghost-8b-beta-GGUF", MessagesFormatterType.MISTRAL],
|
| 21 |
"Honey-Yuzu-13B.Q4_K_M.gguf": ["backyardai/Honey-Yuzu-13B-GGUF", MessagesFormatterType.MISTRAL],
|
|
|
|
| 16 |
"suzume-llama-3-8B-japanese.Q4_K_M.gguf": ["PrunaAI/lightblue-suzume-llama-3-8B-japanese-GGUF-smashed", MessagesFormatterType.LLAMA_3],
|
| 17 |
"suzume-llama-3-8B-multilingual-orpo-borda-top25.Q4_K_M.gguf": ["RichardErkhov/lightblue_-_suzume-llama-3-8B-multilingual-orpo-borda-top25-gguf", MessagesFormatterType.LLAMA_3],
|
| 18 |
"Bungo-L3-8B.Q5_K_M.gguf": ["backyardai/Bungo-L3-8B-GGUF", MessagesFormatterType.LLAMA_3],
|
| 19 |
+
"L3-8B-Tamamo-v1.i1-Q5_K_M.gguf": ["mradermacher/L3-8B-Tamamo-v1-i1-GGUF", MessagesFormatterType.LLAMA_3],
|
| 20 |
"Mistral-Nemo-Instruct-2407-Q4_K_M.gguf": ["bartowski/Mistral-Nemo-Instruct-2407-GGUF", MessagesFormatterType.MISTRAL],
|
| 21 |
"ghost-8b-beta.q5_k.gguf": ["ZeroWw/ghost-8b-beta-GGUF", MessagesFormatterType.MISTRAL],
|
| 22 |
"Honey-Yuzu-13B.Q4_K_M.gguf": ["backyardai/Honey-Yuzu-13B-GGUF", MessagesFormatterType.MISTRAL],
|
tagger.py
CHANGED
|
@@ -62,9 +62,15 @@ def list_uniq(l):
|
|
| 62 |
|
| 63 |
|
| 64 |
def load_dict_from_csv(filename):
|
| 65 |
-
|
| 66 |
-
lines = f.readlines()
|
| 67 |
dict = {}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
for line in lines:
|
| 69 |
parts = line.strip().split(',')
|
| 70 |
dict[parts[0]] = parts[1]
|
|
@@ -242,13 +248,13 @@ def convert_tags_to_ja(input_prompt: str = ""):
|
|
| 242 |
enable_auto_recom_prompt = True
|
| 243 |
|
| 244 |
|
| 245 |
-
animagine_ps = to_list("
|
| 246 |
animagine_nps = to_list("lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]")
|
| 247 |
-
pony_ps = to_list("
|
| 248 |
-
pony_nps = to_list("source_pony,
|
| 249 |
other_ps = to_list("anime artwork, anime style, studio anime, highly detailed, cinematic photo, 35mm photograph, film, bokeh, professional, 4k, highly detailed")
|
| 250 |
other_nps = to_list("photo, deformed, black and white, realism, disfigured, low contrast, drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly")
|
| 251 |
-
default_ps = to_list("
|
| 252 |
default_nps = to_list("score_6, score_5, score_4, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]")
|
| 253 |
def insert_recom_prompt(prompt: str = "", neg_prompt: str = "", type: str = "None"):
|
| 254 |
global enable_auto_recom_prompt
|
|
|
|
| 62 |
|
| 63 |
|
| 64 |
def load_dict_from_csv(filename):
|
| 65 |
+
from pathlib import Path
|
|
|
|
| 66 |
dict = {}
|
| 67 |
+
if not Path(filename).exists(): return dict
|
| 68 |
+
try:
|
| 69 |
+
with open(filename, 'r', encoding="utf-8") as f:
|
| 70 |
+
lines = f.readlines()
|
| 71 |
+
except Exception:
|
| 72 |
+
print(f"Failed to open dictionary file: {filename}")
|
| 73 |
+
return dict
|
| 74 |
for line in lines:
|
| 75 |
parts = line.strip().split(',')
|
| 76 |
dict[parts[0]] = parts[1]
|
|
|
|
| 248 |
enable_auto_recom_prompt = True
|
| 249 |
|
| 250 |
|
| 251 |
+
animagine_ps = to_list("masterpiece, best quality, very aesthetic, absurdres")
|
| 252 |
animagine_nps = to_list("lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]")
|
| 253 |
+
pony_ps = to_list("score_9, score_8_up, score_7_up, masterpiece, best quality, very aesthetic, absurdres")
|
| 254 |
+
pony_nps = to_list("source_pony, score_6, score_5, score_4, busty, ugly face, mutated hands, low res, blurry face, black and white, the simpsons, overwatch, apex legends")
|
| 255 |
other_ps = to_list("anime artwork, anime style, studio anime, highly detailed, cinematic photo, 35mm photograph, film, bokeh, professional, 4k, highly detailed")
|
| 256 |
other_nps = to_list("photo, deformed, black and white, realism, disfigured, low contrast, drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly")
|
| 257 |
+
default_ps = to_list("highly detailed, masterpiece, best quality, very aesthetic, absurdres")
|
| 258 |
default_nps = to_list("score_6, score_5, score_4, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]")
|
| 259 |
def insert_recom_prompt(prompt: str = "", neg_prompt: str = "", type: str = "None"):
|
| 260 |
global enable_auto_recom_prompt
|