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="."):
|