arnaudstiegler commited on
Commit
ab3d555
·
verified ·
1 Parent(s): 8a12e59

End of training

Browse files
README.md ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: CompVis/stable-diffusion-v1-4
3
+ library_name: diffusers
4
+ license: creativeml-openrail-m
5
+ inference: true
6
+ tags:
7
+ - stable-diffusion
8
+ - stable-diffusion-diffusers
9
+ - text-to-image
10
+ - diffusers
11
+ - diffusers-training
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+
18
+ # GameNgen fine-tuning - arnaudstiegler/sd-model-gameNgen
19
+ 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.
20
+
21
+ ![img_0](./image_0.png)
22
+ ![img_1](./image_1.png)
23
+
24
+
25
+
26
+ ## Intended uses & limitations
27
+
28
+ #### How to use
29
+
30
+ ```python
31
+ # TODO: add an example code snippet for running this diffusion pipeline
32
+ ```
33
+
34
+ #### Limitations and bias
35
+
36
+ [TODO: provide examples of latent issues and potential remediations]
37
+
38
+ ## Training details
39
+
40
+ [TODO: describe the data used to train the model]
action_embedding.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:74e743877dbdb33dbf07c0d1ef0bdab1adedc8841d1b8e1b7a389f8f0d23ab51
3
  size 56585
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42f4f8ab029412e0e1f033f6f4144ff04103459a42da9a7ef2290530eab482e7
3
  size 56585
action_embedding_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e39cb8d6aeab27cdc6e0a18b64d841d8ea8227d96d9f75025025cb1bbf695286
3
+ size 55376
image_0.png ADDED
image_1.png ADDED
noise_scheduler/scheduler_config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDIMScheduler",
3
+ "_diffusers_version": "0.31.0",
4
+ "beta_end": 0.012,
5
+ "beta_schedule": "scaled_linear",
6
+ "beta_start": 0.00085,
7
+ "clip_sample": false,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "num_train_timesteps": 1000,
11
+ "prediction_type": "v_prediction",
12
+ "rescale_betas_zero_snr": false,
13
+ "sample_max_value": 1.0,
14
+ "set_alpha_to_one": false,
15
+ "skip_prk_steps": true,
16
+ "steps_offset": 1,
17
+ "thresholding": false,
18
+ "timestep_spacing": "leading",
19
+ "trained_betas": null
20
+ }
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0bd9e15294517671eee92d8f6b58db264a7d5a9a88659b5bbadfcf8aacde65fd
3
- size 153616384
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a4a96853c22eadfae0ac36ad2b22ceae6a5a1c3c081213980065e9625fc37b4
3
+ size 3438633568
vae/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:77b5bf98e9e7c4bbeaaf8dcb2063cf1d7af619de92c0aaecbe4c8adf64d7608f
3
- size 154140672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a47c5b65cf0a57896e2c89f1f8e1050bd209798eeb9f8bf91a4e5612f0b5150a
3
+ size 334643268