DanTagGen delta DanTagGen(Danbooru Tag Generator) is inspired from p1atdev's dart project. But with different arch, dataset, format and different training strategy. Difference between versions alpha: pretrain on 2M dataset, smaller batch size. Limited ability beta: pretrain on 5.3M dataset, larger batch size. More stable, better ability with only a few information provided. delta: pretrain on 7.2M dataset, larger batch size. Slightly underfit but better diversity. quality tag introduced. Model arch This version of DTG is trained from scratch with 400M param LLaMA arch.(In my personal preference I will call it NanoLLaMA) Since it is llama arch. Theoritically it should be able to be used in any LLaMA inference interface. This repo also provided converted FP16 gguf model and quantized 8bit/6bit gguf models. Basically it is recommended to use llama.cpp or llama cpp python to run this model. Which will be very fast. Format for example: And you may get something like: Dataset and Training I use the trainer I implemented in HakuPhi to run the training. with 10epoch on 7.2M data. This model have roughly 10~15B token seen. The dataset is exported by HakuBooru with my danbooru sqlite databas…
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy