FredZhang7
commited on
Commit
•
e279c22
1
Parent(s):
37cc880
Added missing words
Browse files
README.md
CHANGED
@@ -22,7 +22,7 @@ datasets:
|
|
22 |
## Fast Anime PromptGen
|
23 |
|
24 |
The main model (`pytorch_model.bin`) was trained on a dataset of **80,000** anime tags and for 3 epochs. I fetched the tags from the [Safebooru API endpoint](https://safebooru.donmai.us/posts/random.json), but only accepted the ones with **up_score ≥ 8** and without any [blacklisted tags](./blacklist.txt).
|
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 in the training data.
|
26 |
Here's the complete [prompt preprocessing algorithm](./preprocess.py).
|
27 |
|
28 |
|
|
|
22 |
## Fast Anime PromptGen
|
23 |
|
24 |
The main model (`pytorch_model.bin`) was trained on a dataset of **80,000** anime tags and for 3 epochs. I fetched the tags from the [Safebooru API endpoint](https://safebooru.donmai.us/posts/random.json), but only accepted the ones with **up_score ≥ 8** and without any [blacklisted tags](./blacklist.txt).
|
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 pipeline params, model structure or training duration, but rather from the random usernames in the training data.
|
26 |
Here's the complete [prompt preprocessing algorithm](./preprocess.py).
|
27 |
|
28 |
|