Commit f48fd2de authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add embedding support

parent 2b862b6b

imgutils/tagging/wd14.py

+1 −1
Original line number Diff line number Diff line
@@ -231,6 +231,6 @@ def get_wd14_tags( 
            'general': general_res,

            'character': character_res,

            'tag': {**general_res, **character_res},

            'embedding': embeddings,

            'embedding': embeddings[0],

        }

    )