giannisdaras commited on
Commit
862279d
·
verified ·
1 Parent(s): a48c1bc

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +25 -0
README.md ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - ddpo
5
+ - stable-diffusion
6
+ - reinforcement-learning
7
+ - lora
8
+ ---
9
+
10
+ # DDPO Sharpness Checkpoints
11
+
12
+ LoRA checkpoints from DDPO fine-tuning of Stable Diffusion v1.4 with a sharpness reward (Laplacian variance).
13
+
14
+ ## Training Details
15
+
16
+ - Base model: CompVis/stable-diffusion-v1-4
17
+ - Method: DDPO with LoRA
18
+ - Reward: Sharpness (Laplacian variance)
19
+ - Prompts: ImageNet animals
20
+ - Epochs: 100
21
+ - Compute: 4x A100-SXM4-80GB (MIT ORCD)
22
+
23
+ ## Checkpoints
24
+
25
+ Uploaded every 10 epochs: checkpoint_0, checkpoint_10, ..., checkpoint_90, checkpoint_98 (final).