mbsdeepak commited on
Commit
1f74f4d
·
verified ·
1 Parent(s): 61efe3c

Update to guidance 1.5 (better samples)

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -23,7 +23,7 @@ from a caption using **classifier-free guidance**.
23
  ![samples](sample.png)
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 3.0.*
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 3.0
 
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
  ![samples](sample.png)
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.