--- base_model: CompVis/stable-diffusion-v1-4 library_name: diffusers license: creativeml-openrail-m inference: true tags: - stable-diffusion - stable-diffusion-diffusers - text-to-image - diffusers - diffusers-training - stable-diffusion - stable-diffusion-diffusers - text-to-image - diffusers - diffusers-training --- # 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. ![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) ## Intended uses & limitations #### How to use See: https://github.com/arnaudstiegler/gameNgen-repro/blob/main/README.md ## Training details 20k steps batch size 12 10 epochs with CFG data augmentation