change path for imports
Browse files- handler.py +18 -18
handler.py
CHANGED
|
@@ -29,33 +29,33 @@ if device.type != "cuda":
|
|
| 29 |
|
| 30 |
class EndpointHandler:
|
| 31 |
LORA_PATHS = {
|
| 32 |
-
"hairdetailer": "hairdetailer.safetensors",
|
| 33 |
-
"lora_leica": "lora_leica.safetensors",
|
| 34 |
-
"epiNoiseoffset_v2": "epiNoiseoffset_v2.safetensors",
|
| 35 |
-
"MBHU-TT2FRS": "MBHU-TT2FRS.safetensors",
|
| 36 |
-
"ShinyOiledSkin_v20": "ShinyOiledSkin_v20-LoRA.safetensors",
|
| 37 |
-
"polyhedron_new_skin_v1.1": "polyhedron_new_skin_v1.1.safetensors",
|
| 38 |
-
"detailed_eye-10": "detailed_eye-10.safetensors",
|
| 39 |
-
"add_detail": "add_detail.safetensors",
|
| 40 |
-
"MuscleGirl_v1": "MuscleGirl_v1.safetensors",
|
| 41 |
}
|
| 42 |
|
| 43 |
TEXTUAL_INVERSION = [
|
| 44 |
-
# {
|
| 45 |
-
# "weight_name": "EasyNegative.safetensors",
|
| 46 |
-
# "token": "easynegative",
|
| 47 |
-
# },
|
| 48 |
-
{"weight_name": "badhandv4.pt", "token": "badhandv4"},
|
| 49 |
{
|
| 50 |
-
"weight_name": "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
"token": "bad-artist-anime",
|
| 52 |
},
|
| 53 |
-
{"weight_name": "NegfeetV2.pt", "token": "NegfeetV2"},
|
| 54 |
{
|
| 55 |
-
"weight_name": "ng_deepnegative_v1_75t.pt",
|
| 56 |
"token": "ng_deepnegative_v1_75t",
|
| 57 |
},
|
| 58 |
-
{"weight_name": "bad-hands-5.pt", "token": "bad-hands-5"},
|
| 59 |
]
|
| 60 |
|
| 61 |
def __init__(self, path="."):
|
|
|
|
| 29 |
|
| 30 |
class EndpointHandler:
|
| 31 |
LORA_PATHS = {
|
| 32 |
+
"hairdetailer": "isatis/kw/lora/hairdetailer.safetensors",
|
| 33 |
+
"lora_leica": "isatis/kw/lora/lora_leica.safetensors",
|
| 34 |
+
"epiNoiseoffset_v2": "isatis/kw/lora/epiNoiseoffset_v2.safetensors",
|
| 35 |
+
"MBHU-TT2FRS": "isatis/kw/lora/MBHU-TT2FRS.safetensors",
|
| 36 |
+
"ShinyOiledSkin_v20": "isatis/kw/lora/ShinyOiledSkin_v20-LoRA.safetensors",
|
| 37 |
+
"polyhedron_new_skin_v1.1": "isatis/kw/lora/polyhedron_new_skin_v1.1.safetensors",
|
| 38 |
+
"detailed_eye-10": "isatis/kw/lora/detailed_eye-10.safetensors",
|
| 39 |
+
"add_detail": "isatis/kw/lora/add_detail.safetensors",
|
| 40 |
+
"MuscleGirl_v1": "isatis/kw/lora/MuscleGirl_v1.safetensors",
|
| 41 |
}
|
| 42 |
|
| 43 |
TEXTUAL_INVERSION = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
{
|
| 45 |
+
"weight_name": "https://huggingface.co/isatis/kw/embeddings/EasyNegative.safetensors",
|
| 46 |
+
"token": "easynegative",
|
| 47 |
+
},
|
| 48 |
+
{"weight_name": "isatis/kw/embeddings/badhandv4.pt", "token": "badhandv4"},
|
| 49 |
+
{
|
| 50 |
+
"weight_name": "isatis/kw/embeddings/bad-artist-anime.pt",
|
| 51 |
"token": "bad-artist-anime",
|
| 52 |
},
|
| 53 |
+
{"weight_name": "isatis/kw/embeddings/NegfeetV2.pt", "token": "NegfeetV2"},
|
| 54 |
{
|
| 55 |
+
"weight_name": "isatis/kw/embeddings/ng_deepnegative_v1_75t.pt",
|
| 56 |
"token": "ng_deepnegative_v1_75t",
|
| 57 |
},
|
| 58 |
+
{"weight_name": "isatis/kw/embeddings/bad-hands-5.pt", "token": "bad-hands-5"},
|
| 59 |
]
|
| 60 |
|
| 61 |
def __init__(self, path="."):
|