├── README.md ├── main.py ├── requirements.txt ├── run.sh ├── sm.png ├── sm_cover.png ├── src ├── LICENSE ├── __init__.py ├── dataset.py ├── model.py ├── modeling_llama.py ├── setup.py ├── tokenizer.json ├── trainer.py └── utils.py └── test_environment.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/README.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/run.sh -------------------------------------------------------------------------------- /sm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/sm.png -------------------------------------------------------------------------------- /sm_cover.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/sm_cover.png -------------------------------------------------------------------------------- /src/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/LICENSE -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/model.py -------------------------------------------------------------------------------- /src/modeling_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/modeling_llama.py -------------------------------------------------------------------------------- /src/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/setup.py -------------------------------------------------------------------------------- /src/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/tokenizer.json -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/trainer.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/src/utils.py -------------------------------------------------------------------------------- /test_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MikaStars39/StableMask/HEAD/test_environment.py --------------------------------------------------------------------------------