multimodalart
/

fofr-sdxl-emoji

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

multimodalart HF staff commited on Jun 26

Commit

6fcafb6

•

1 Parent(s): b3d9998

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -30,8 +30,8 @@ class EndpointHandler:
         )
         state_dict = load_file(embedding_path)
-        self.pipe.load_textual_inversion(state_dict["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=self.pipe.text_encoder, tokenizer=self.pipe.tokenizer)
-        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)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         """

         )
         state_dict = load_file(embedding_path)
+        self.pipe.load_textual_inversion(state_dict["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=self.pipe.text_encoder, tokenizer=self.pipe.tokenizer)
+        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)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         """