Update README.md

README.md

@@ -20,11 +20,12 @@ tags:
 should probably proofread and complete it, then remove this comment. -->
 
 
-# GameNgen fine-tuning - arnaudstiegler/sd-model-gameNgen
+# GameNgen fine-tuning - arnaudstiegler/gameNgen-baseline-20ksteps
 Full finetune of CompVis/stable-diffusion-v1-4. The weights were fine-tuned on the P-H-B-D-a16z/ViZDoom-Deathmatch-PPO-Lrg dataset. You can find some example images in the following. 
 
-![img_0](./image_0.png)
-![img_1](./image_1.png)
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/600997fb731233b499672cbd/iTGZ2z-62z0qCXs0UOaxg.png)
+![image/gif](https://cdn-uploads.huggingface.co/production/uploads/600997fb731233b499672cbd/BVTpTk9lUrurmOgdr3NA2.gif)
 
 
 
@@ -32,14 +33,11 @@ Full finetune of CompVis/stable-diffusion-v1-4. The weights were fine-tuned on t
 
 #### How to use
 
-```python
-# TODO: add an example code snippet for running this diffusion pipeline
-```
-
-#### Limitations and bias
-
-[TODO: provide examples of latent issues and potential remediations]
+See: https://github.com/arnaudstiegler/gameNgen-repro/blob/main/README.md
 
 ## Training details
 
-[TODO: describe the data used to train the model]
+20k steps
+batch size 12
+10 epochs
+with CFG data augmentation