Commit
•
5d7f667
1
Parent(s):
6fcafb6
Update handler.py
Browse files- handler.py +3 -3
handler.py
CHANGED
@@ -23,15 +23,15 @@ class EndpointHandler:
|
|
23 |
).to(device)
|
24 |
|
25 |
self.pipe.load_lora_weights("SvenN/sdxl-emoji", weight_name="lora.safetensors")
|
26 |
-
self.pipe.fuse_lora()
|
27 |
|
28 |
embedding_path = hf_hub_download(
|
29 |
repo_id="SvenN/sdxl-emoji", filename="embeddings.pti", repo_type="model"
|
30 |
)
|
31 |
state_dict = load_file(embedding_path)
|
32 |
|
33 |
-
self.pipe.load_textual_inversion(state_dict["
|
34 |
-
self.pipe.load_textual_inversion(state_dict["
|
35 |
|
36 |
def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
|
37 |
"""
|
|
|
23 |
).to(device)
|
24 |
|
25 |
self.pipe.load_lora_weights("SvenN/sdxl-emoji", weight_name="lora.safetensors")
|
26 |
+
self.pipe.fuse_lora(lora_scale=0.6)
|
27 |
|
28 |
embedding_path = hf_hub_download(
|
29 |
repo_id="SvenN/sdxl-emoji", filename="embeddings.pti", repo_type="model"
|
30 |
)
|
31 |
state_dict = load_file(embedding_path)
|
32 |
|
33 |
+
self.pipe.load_textual_inversion(state_dict["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=self.pipe.text_encoder, tokenizer=self.pipe.tokenizer)
|
34 |
+
self.pipe.load_textual_inversion(state_dict["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=self.pipe.text_encoder_2, tokenizer=self.pipe.tokenizer_2)
|
35 |
|
36 |
def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
|
37 |
"""
|