Update README.md
Browse files
README.md
CHANGED
|
@@ -2,4 +2,10 @@
|
|
| 2 |
license: mit
|
| 3 |
---
|
| 4 |
|
| 5 |
-
models are diffusion trainers from https://github.com/lucidrains/DALLE2-pytorch with defaults specified in the train_diffusion_prior.py script
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: mit
|
| 3 |
---
|
| 4 |
|
| 5 |
+
models are diffusion trainers from https://github.com/lucidrains/DALLE2-pytorch with defaults specified in the train_diffusion_prior.py script
|
| 6 |
+
|
| 7 |
+
notably:
|
| 8 |
+
|
| 9 |
+
depth=12
|
| 10 |
+
d_model=768
|
| 11 |
+
clip = OpenAIClipAdapter(clip_choice=["ViT-L/14" | "ViT-B/32"]
|