1580158
/
stable-diffusion-webui-dataset-tag-editor
mirror of https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update waifu_diffusion_tagger_timm.py 

fix #100
main
toshiaki1729 2024-05-26 13:17:17 +09:00
parent 94cc5a2a6b
commit da5053bc2e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/dataset_tag_editor/interrogators/waifu_diffusion_tagger_timm.py
View File

@ -81,7 +81,7 @@ class WaifuDiffusionTaggerTimm:
with torch.inference_mode():
features = self.model.forward(image_t)
probs = F.sigmoid(features).detach().cpu()
probs = F.sigmoid(features).detach().cpu().numpy()
labels: list[Tuple[str, float]] = list(zip(self.labels, probs[0].astype(float)))