yh0317777
/

Altclip_fork

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

Yiheng Hu commited on Oct 2, 2024

Commit

b44ea03

·

1 Parent(s): 25162b9

update

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -27,4 +27,4 @@ class EndpointHandler():
             texts = data.pop("inputs",data)
             inputs = self.md_processor(text = texts, padding=True, return_tensors="pt").to(device)
             text_feature = self.md_model.get_text_features(**inputs)
-            return {"feature":text_feature.cpu().numpy()}

             texts = data.pop("inputs",data)
             inputs = self.md_processor(text = texts, padding=True, return_tensors="pt").to(device)
             text_feature = self.md_model.get_text_features(**inputs)
+            return {"feature":text_feature.cpu().tolist()}