FredZhang7 commited on
Commit
81017b5
1 Parent(s): 3da8a05

Update data preprocessing

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
- `pytorch_model` is trained on 80K anime tags, all with `up_score` greater than 8 and without "greyscale","girls","boys", and "others", fetched from the [Safebooru API](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 "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