julien-c
/

fofr-sdxl-emoji

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

multimodalart HF staff commited on Jun 25

Commit

1983d3d

•

1 Parent(s): 8f21bb2

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=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
-        self.pipe.load_textual_inversion(state_dict["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.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_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]]]:
         """