Upload 7 files

llmdolphin.py

@@ -18,6 +18,12 @@ llm_models = {
     "Lumimaid-Magnum-12B.i1-Q4_K_M.gguf": ["mradermacher/Lumimaid-Magnum-12B-i1-GGUF", MessagesFormatterType.MISTRAL],
     "Nemo-12B-Marlin-v1.i1-Q4_K_M.gguf": ["mradermacher/Nemo-12B-Marlin-v1-i1-GGUF", MessagesFormatterType.MISTRAL],
     "Nemo-12B-Marlin-v2.i1-Q4_K_M.gguf": ["mradermacher/Nemo-12B-Marlin-v2-i1-GGUF", MessagesFormatterType.MISTRAL],
+    "Nemo-12B-Marlin-v3.Q4_K_M.gguf": ["mradermacher/Nemo-12B-Marlin-v3-GGUF", MessagesFormatterType.MISTRAL],
+    "MN-12B-Lyra-v1.i1-Q4_K_M.gguf": ["mradermacher/MN-12B-Lyra-v1-i1-GGUF", MessagesFormatterType.CHATML],
+    "MN-12B-Starsong-v1.i1-Q4_K_M.gguf": ["mradermacher/MN-12B-Starsong-v1-i1-GGUF", MessagesFormatterType.CHATML],
+    "Humanish-Roleplay-Llama-3.1-8B.i1-Q5_K_M.gguf": ["mradermacher/Humanish-Roleplay-Llama-3.1-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
+    "Llama-3-Luminurse-v0.1-OAS-8B.Q5_K_M.gguf": ["grimjim/Llama-3-Luminurse-v0.1-OAS-8B-GGUF", MessagesFormatterType.LLAMA_3],
+    "L3.1-8B-Niitama-v1.1-Q5_K_M-imat.gguf": ["L3.1-8B-Niitama-v1.1-Q5_K_M-imat.gguf", MessagesFormatterType.MISTRAL],
     "Evolved-Llama3-8B.i1-Q5_K_M.gguf": ["mradermacher/Evolved-Llama3-8B-i1-GGUF", MessagesFormatterType.LLAMA_3],
     "DarkIdol-Llama-3.1-8B-Instruct-1.2-Uncensored-Q5_K_M.gguf": ["bartowski/DarkIdol-Llama-3.1-8B-Instruct-1.2-Uncensored-GGUF", MessagesFormatterType.LLAMA_3],
     "Llama-3-Swallow-8B-Instruct-v0.1.Q5_K_M.gguf": ["YukiTomita-CC/Llama-3-Swallow-8B-Instruct-v0.1-IMat-GGUF_dolly-15k-ja-prompt", MessagesFormatterType.ALPACA],

tagger.py

@@ -6,6 +6,7 @@ from transformers import (
     AutoImageProcessor,
     AutoModelForImageClassification,
 )
+from pathlib import Path
 
 
 WD_MODEL_NAMES = ["p1atdev/wd-swinv2-tagger-v3-hf"]
@@ -30,12 +31,15 @@ PEOPLE_TAGS = (
 
 
 RATING_MAP = {
+    "sfw": "safe",
     "general": "safe",
     "sensitive": "sensitive",
     "questionable": "nsfw",
     "explicit": "explicit, nsfw",
 }
 DANBOORU_TO_E621_RATING_MAP = {
+    "sfw": "rating_safe",
+    "general": "rating_safe",
     "safe": "rating_safe",
     "sensitive": "rating_safe",
     "nsfw": "rating_explicit",
@@ -90,9 +94,10 @@ def list_uniq(l):
 
 
 def load_dict_from_csv(filename):
-    from pathlib import Path
     dict = {}
-    if not Path(filename).exists(): return dict
+    if not Path(filename).exists():
+        if Path('./tagger/', filename).exists(): filename = str(Path('./tagger/', filename))
+        else: return dict
     try:
         with open(filename, 'r', encoding="utf-8") as f:
             lines = f.readlines()
@@ -128,7 +133,8 @@ def character_list_to_series_list(character_list):
 
 
 def select_random_character(series: str, character: str):
-    from random import randrange
+    from random import seed, randrange
+    seed()
     character_list = list(anime_series_dict.keys())
     character = character_list[randrange(len(character_list) - 1)]
     series = anime_series_dict.get(character.split(",")[0].strip(), "")
@@ -146,7 +152,6 @@ def danbooru_to_e621(dtag, e621_dict):
     
     import re
     tag = re.sub(r'[\w ]+', lambda wrapper: d_to_e(wrapper, e621_dict), dtag, 2)
-
     return tag
 
 
@@ -317,6 +322,7 @@ def insert_recom_prompt(prompt: str = "", neg_prompt: str = "", type: str = "Non
 def load_model_prompt_dict():
     import json
     dict = {}
+    path = 'model_dict.json' if Path('model_dict.json').exists() else './tagger/model_dict.json'
     try:
         with open('model_dict.json', encoding='utf-8') as f:
             dict = json.load(f)

utils.py

@@ -43,3 +43,8 @@ COPY_ACTION_JS = """\
     navigator.clipboard.writeText(inputs);
   }
 }"""
+
+
+def gradio_copy_prompt(prompt: str):
+    gr.Info("Copied!")
+    return prompt