Update to guidance 1.5 (better samples)
Browse files
README.md
CHANGED
|
@@ -23,7 +23,7 @@ from a caption using **classifier-free guidance**.
|
|
| 23 |

|
| 24 |
|
| 25 |
*One row per class (t-shirt, trouser, pullover, dress, coat, sandal, shirt, sneaker, bag, ankle
|
| 26 |
-
boot); each image is generated from pure noise, DDIM 50 steps, guidance
|
| 27 |
|
| 28 |
## What it does
|
| 29 |
|
|
@@ -83,7 +83,8 @@ save_image(denormalize(imgs), "out.png", nrow=len(FASHION_CLASSES))
|
|
| 83 |
- **Conditioning:** frozen CLIP (`openai/clip-vit-base-patch32`) text embeddings via FiLM +
|
| 84 |
cross-attention; 15% caption dropout for classifier-free guidance
|
| 85 |
- **Trained:** 15 epochs on Apple Silicon (MPS), final loss ≈ 0.043
|
| 86 |
-
- **Sampling:** DDIM, 50 steps, guidance scale
|
|
|
|
| 87 |
|
| 88 |
> These are the **raw** weights, not EMA — for a short (15-epoch) run the EMA average still lags
|
| 89 |
> the live weights, so the raw model produces the cleaner samples.
|
|
|
|
| 23 |

|
| 24 |
|
| 25 |
*One row per class (t-shirt, trouser, pullover, dress, coat, sandal, shirt, sneaker, bag, ankle
|
| 26 |
+
boot); each image is generated from pure noise, DDIM 50 steps, guidance 1.5.*
|
| 27 |
|
| 28 |
## What it does
|
| 29 |
|
|
|
|
| 83 |
- **Conditioning:** frozen CLIP (`openai/clip-vit-base-patch32`) text embeddings via FiLM +
|
| 84 |
cross-attention; 15% caption dropout for classifier-free guidance
|
| 85 |
- **Trained:** 15 epochs on Apple Silicon (MPS), final loss ≈ 0.043
|
| 86 |
+
- **Sampling:** DDIM, 50 steps, guidance scale 1.5 (low guidance — high CFG artifacts on this
|
| 87 |
+
small, short-trained model)
|
| 88 |
|
| 89 |
> These are the **raw** weights, not EMA — for a short (15-epoch) run the EMA average still lags
|
| 90 |
> the live weights, so the raw model produces the cleaner samples.
|