FredZhang7 commited on
Commit
3fac12e
1 Parent(s): 0eef561

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ datasets:
21
 
22
  ## Fast Anime PromptGen
23
 
24
- The main model `pytorch_model.bin` is trained on **80K** anime tags, all with **up_score ≥ 8** and without "greyscale","girls","boys", and "others", fetched from the [Safebooru API endpoint](https://safebooru.donmai.us/posts/random.json).
25
  I didn't release the V1 model because it only generated gibberish prompts. After trying all means to correct that behavior, I eventually figured that the cause of the gibberish prompts is not from the model or training duration, but rather from the random usernames present in the training data.
26
  Here's the complete [prompt preprocessing](./preprocess.py).
27
 
 
21
 
22
  ## Fast Anime PromptGen
23
 
24
+ The main model `pytorch_model.bin` is trained on **80K** anime tags, all with **up_score ≥ 8** and without the [blacklisted tags](./blacklist.txt), fetched from the [Safebooru API endpoint](https://safebooru.donmai.us/posts/random.json).
25
  I didn't release the V1 model because it only generated gibberish prompts. After trying all means to correct that behavior, I eventually figured that the cause of the gibberish prompts is not from the model or training duration, but rather from the random usernames present in the training data.
26
  Here's the complete [prompt preprocessing](./preprocess.py).
27