Muhammadreza commited on
Commit
c7e6548
1 Parent(s): 7bc1ec0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -13,4 +13,9 @@ library_name: diffusers
13
 
14
  ### Training process
15
 
16
- _TBD_
 
 
 
 
 
 
13
 
14
  ### Training process
15
 
16
+ The code for pre-processing data and fine-tuning the model is available in [this repository](https://github.com/prp-e/mann-e_training) and you can run it on your own as well.
17
+
18
+ * Text encoder iterations : 1440 (number of pics times two in order to understand `mstyle` which can give the user a _Midjourney version 5_ vibe).
19
+ * Stable Diffusion iterations : 16000 iterations for one epoch
20
+ * Time: around 4 hours on a single T4 GPU.
21
+