change path for imports

handler.py

@@ -20,6 +20,13 @@ from torch import autocast
 # https://huggingface.co/docs/inference-endpoints/guides/custom_handler
 
 
+# if local avoid repo url
+LOCAL = False
+PREFIX_URL = ""
+if not LOCAL:
+    PREFIX_URL = "https://huggingface.co/isatis/kw/"
+
+
 # set device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -29,33 +36,45 @@ if device.type != "cuda":
 
 class EndpointHandler:
     LORA_PATHS = {
-        "hairdetailer": "isatis/kw/lora/hairdetailer.safetensors",
-        "lora_leica": "isatis/kw/lora/lora_leica.safetensors",
-        "epiNoiseoffset_v2": "isatis/kw/lora/epiNoiseoffset_v2.safetensors",
-        "MBHU-TT2FRS": "isatis/kw/lora/MBHU-TT2FRS.safetensors",
-        "ShinyOiledSkin_v20": "isatis/kw/lora/ShinyOiledSkin_v20-LoRA.safetensors",
-        "polyhedron_new_skin_v1.1": "isatis/kw/lora/polyhedron_new_skin_v1.1.safetensors",
-        "detailed_eye-10": "isatis/kw/lora/detailed_eye-10.safetensors",
-        "add_detail": "isatis/kw/lora/add_detail.safetensors",
-        "MuscleGirl_v1": "isatis/kw/lora/MuscleGirl_v1.safetensors",
+        "hairdetailer": PREFIX_URL + "lora/hairdetailer.safetensors",
+        "lora_leica": PREFIX_URL + "lora/lora_leica.safetensors",
+        "epiNoiseoffset_v2": PREFIX_URL + "lora/epiNoiseoffset_v2.safetensors",
+        "MBHU-TT2FRS": PREFIX_URL + "lora/MBHU-TT2FRS.safetensors",
+        "ShinyOiledSkin_v20": PREFIX_URL + "lora/ShinyOiledSkin_v20-LoRA.safetensors",
+        "polyhedron_new_skin_v1.1": PREFIX_URL
+        + "lora/polyhedron_new_skin_v1.1.safetensors",
+        "detailed_eye-10": PREFIX_URL + "lora/detailed_eye-10.safetensors",
+        "add_detail": PREFIX_URL + "lora/add_detail.safetensors",
+        "MuscleGirl_v1": PREFIX_URL + "lora/MuscleGirl_v1.safetensors",
     }
 
     TEXTUAL_INVERSION = [
         {
-            "weight_name": "https://huggingface.co/isatis/kw/embeddings/EasyNegative.safetensors",
+            "weight_name": PREFIX_URL + "embeddings/EasyNegative.safetensors",
             "token": "easynegative",
         },
-        {"weight_name": "isatis/kw/embeddings/badhandv4.pt", "token": "badhandv4"},
         {
-            "weight_name": "isatis/kw/embeddings/bad-artist-anime.pt",
+            "weight_name": PREFIX_URL + "embeddings/EasyNegative.safetensors",
+            "token": "EasyNegative",
+        },
+        {"weight_name": PREFIX_URL + "embeddings/badhandv4.pt", "token": "badhandv4"},
+        {
+            "weight_name": PREFIX_URL + "embeddings/bad-artist-anime.pt",
             "token": "bad-artist-anime",
         },
-        {"weight_name": "isatis/kw/embeddings/NegfeetV2.pt", "token": "NegfeetV2"},
+        {"weight_name": PREFIX_URL + "embeddings/NegfeetV2.pt", "token": "NegfeetV2"},
         {
-            "weight_name": "isatis/kw/embeddings/ng_deepnegative_v1_75t.pt",
+            "weight_name": PREFIX_URL + "embeddings/ng_deepnegative_v1_75t.pt",
             "token": "ng_deepnegative_v1_75t",
         },
-        {"weight_name": "isatis/kw/embeddings/bad-hands-5.pt", "token": "bad-hands-5"},
+        {
+            "weight_name": PREFIX_URL + "embeddings/ng_deepnegative_v1_75t.pt",
+            "token": "NG_DeepNegative_V1_75T",
+        },
+        {
+            "weight_name": PREFIX_URL + "embeddings/bad-hands-5.pt",
+            "token": "bad-hands-5",
+        },
     ]
 
     def __init__(self, path="."):