multimodalart HF staff commited on
Commit
5d7f667
1 Parent(s): 6fcafb6

Update handler.py

Browse files
Files changed (1) hide show
  1. 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["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=self.pipe.text_encoder, tokenizer=self.pipe.tokenizer)
34
- self.pipe.load_textual_inversion(state_dict["text_encoders_0"], 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
  """
 
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
  """