Using PAth for imports

handler.py

@@ -6,6 +6,7 @@ from io import BytesIO
 from pprint import pprint
 from typing import Any, Dict, List
 import os
+from pathlib import Path
 
 import torch
 from diffusers import (
@@ -20,13 +21,10 @@ from torch import autocast
 # https://huggingface.co/philschmid/stable-diffusion-v1-4-endpoints
 # https://huggingface.co/docs/inference-endpoints/guides/custom_handler
 
+REPO_DIR = Path(__file__).resolve().parent
 
 # if local avoid repo url
-LOCAL = False
-PREFIX_URL = ""
-if not LOCAL:
-    PREFIX_URL = "https://huggingface.co/isatis/kw/tree/main/"
-
+print(os.getcwd())
 
 # set device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -36,46 +34,53 @@ if device.type != "cuda":
 
 
 class EndpointHandler:
-    print(os.getcwd())
-
     LORA_PATHS = {
-        "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",
+        "hairdetailer": str(REPO_DIR / "lora/hairdetailer.safetensors"),
+        "lora_leica": str(REPO_DIR / "lora/lora_leica.safetensors"),
+        "epiNoiseoffset_v2": str(REPO_DIR / "lora/epiNoiseoffset_v2.safetensors"),
+        "MBHU-TT2FRS": str(REPO_DIR / "lora/MBHU-TT2FRS.safetensors"),
+        "ShinyOiledSkin_v20": str(
+            REPO_DIR / "lora/ShinyOiledSkin_v20-LoRA.safetensors"
+        ),
+        "polyhedron_new_skin_v1.1": str(
+            REPO_DIR / "lora/polyhedron_new_skin_v1.1.safetensors"
+        ),
+        "detailed_eye-10": str(REPO_DIR / "lora/detailed_eye-10.safetensors"),
+        "add_detail": str(REPO_DIR / "lora/add_detail.safetensors"),
+        "MuscleGirl_v1": str(REPO_DIR / "lora/MuscleGirl_v1.safetensors"),
     }
 
     TEXTUAL_INVERSION = [
         {
-            "weight_name": PREFIX_URL + "embeddings/EasyNegative.safetensors",
+            "weight_name": str(REPO_DIR / "embeddings/EasyNegative.safetensors"),
             "token": "easynegative",
         },
         {
-            "weight_name": PREFIX_URL + "embeddings/EasyNegative.safetensors",
+            "weight_name": str(REPO_DIR / "embeddings/EasyNegative.safetensors"),
             "token": "EasyNegative",
         },
-        {"weight_name": PREFIX_URL + "embeddings/badhandv4.pt", "token": "badhandv4"},
         {
-            "weight_name": PREFIX_URL + "embeddings/bad-artist-anime.pt",
+            "weight_name": str(REPO_DIR / "embeddings/badhandv4.pt"),
+            "token": "badhandv4",
+        },
+        {
+            "weight_name": str(REPO_DIR / "embeddings/bad-artist-anime.pt"),
             "token": "bad-artist-anime",
         },
-        {"weight_name": PREFIX_URL + "embeddings/NegfeetV2.pt", "token": "NegfeetV2"},
         {
-            "weight_name": PREFIX_URL + "embeddings/ng_deepnegative_v1_75t.pt",
+            "weight_name": str(REPO_DIR / "embeddings/NegfeetV2.pt"),
+            "token": "NegfeetV2",
+        },
+        {
+            "weight_name": str(REPO_DIR / "embeddings/ng_deepnegative_v1_75t.pt"),
             "token": "ng_deepnegative_v1_75t",
         },
         {
-            "weight_name": PREFIX_URL + "embeddings/ng_deepnegative_v1_75t.pt",
+            "weight_name": str(REPO_DIR / "embeddings/ng_deepnegative_v1_75t.pt"),
             "token": "NG_DeepNegative_V1_75T",
         },
         {
-            "weight_name": PREFIX_URL + "embeddings/bad-hands-5.pt",
+            "weight_name": str(REPO_DIR / "embeddings/bad-hands-5.pt"),
             "token": "bad-hands-5",
         },
     ]