Unverified Commit 5f846655 authored by Naomi Rue Golding's avatar Naomi Rue Golding Committed by GitHub
Browse files

Merge pull request #167 from deepghs/dev/pixai-tagger 

dev(narugo): add support for pixai taggers
parents 74bba37b e9af3868

.github/workflows/doc.yml

+2 −0
Original line number Diff line number Diff line
@@ -67,6 +67,7 @@ jobs: 
          ENV_PROD: 'true'

          PLANTUML_HOST: http://localhost:18080

          HF_TOKEN: ${{ secrets.HF_TOKEN }}

          CI: 'true'

        run: |

          plantumlcli -c

          make docs
@@ -145,6 +146,7 @@ jobs: 
          ENV_PROD: 'true'

          PLANTUML_HOST: http://localhost:18080

          HF_TOKEN: ${{ secrets.HF_TOKEN }}

          CI: 'true'

        run: |

          git fetch --all --tags

          git branch -av

Makefile

+1 −1
Original line number Diff line number Diff line
@@ -41,7 +41,7 @@ unittest: 
		--cov="${RANGE_SRC_DIR}" \

		$(if ${MIN_COVERAGE},--cov-fail-under=${MIN_COVERAGE},) \

		$(if ${WORKERS},-n ${WORKERS},) \

		--reruns 8 --reruns-delay 2 --only-rerun '(OSError|Timeout|HTTPError.*429|HTTPError.*502|HTTPError.*504|check your connection|429 error)'

		--reruns 8 --reruns-delay 2 --only-rerun '(OSError|Timeout|HTTPError.*429|HTTPError.*502|HTTPError.*504|check your connection|429 error|CAS service error)'



docs:

	$(MAKE) -C "${DOC_DIR}" build

docs/source/_libs/benchmark.py

+4 −0
Original line number Diff line number Diff line
@@ -34,6 +34,9 @@ class BaseBenchmark: 
    def unload(self):

        raise NotImplementedError



    def after_unload(self):

        pass



    def run(self):

        raise NotImplementedError
@@ -60,6 +63,7 @@ class BaseBenchmark: 


        self.unload()

        _record('<unload>')

        self.after_unload()



        mems = np.array([mem for _, mem, _ in logs])

        mems -= mems[0]

docs/source/api_doc/tagging/index.rst

+1 −0
Original line number Diff line number Diff line
@@ -12,6 +12,7 @@ imgutils.tagging 
    mldanbooru

    wd14

    camie

    pixai

    deepdanbooru

    deepgelbooru

    format

docs/source/api_doc/tagging/pixai.rst

0 → 100644
+15 −0
Original line number Diff line number Diff line 
imgutils.tagging.pixai

====================================



.. currentmodule:: imgutils.tagging.pixai



.. automodule:: imgutils.tagging.pixai





get_pixai_tags

----------------------



.. autofunction:: get_pixai_tags