Upload 8 files

tagger/fl2flux.py

@@ -8,8 +8,8 @@ import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-fl_model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-Flux', trust_remote_code=True).to("cpu").eval()
-fl_processor = AutoProcessor.from_pretrained('gokaygokay/Florence-2-Flux', trust_remote_code=True)
+fl_model = AutoModelForCausalLM.from_pretrained('gokaygokay/Florence-2-Flux-Large', trust_remote_code=True).to("cpu").eval()
+fl_processor = AutoProcessor.from_pretrained('gokaygokay/Florence-2-Flux-Large', trust_remote_code=True)
 
 
 def fl_modify_caption(caption: str) -> str:

tagger/tagger.py

@@ -189,7 +189,7 @@ def convert_danbooru_to_e621_prompt(input_prompt: str = "", prompt_type: str = "
 
 from translatepy import Translator
 translator = Translator()
-def translate_prompt(prompt: str = ""):
+def translate_prompt_old(prompt: str = ""):
     def translate_to_english(input: str):
         try:
             output = str(translator.translate(input, 'English'))
@@ -218,6 +218,15 @@ def translate_prompt(prompt: str = ""):
     return ", ".join(outputs)
 
 
+def translate_prompt(input: str):
+    try:
+        output = str(translator.translate(input, 'English'))
+    except Exception as e:
+        output = input
+        print(e)
+    return output
+
+
 def translate_prompt_to_ja(prompt: str = ""):
     def translate_to_japanese(input: str):
         try: