Commit 138ada24 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): try fix unittest, ci skip

parent e7b8cd15
Loading
Loading
Loading
Loading
+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
+3 −1
Original line number Diff line number Diff line
import os
import random

from hfutils.cache import delete_cache
@@ -9,6 +10,7 @@ from imgutils.tagging import get_deepdanbooru_tags, get_wd14_tags, get_mldanboor

class CleanModelStorageBenchmark(BaseBenchmark):
    def after_unload(self):
        if os.environ.get('CI'):
            delete_cache()