GamerC0der commited on
Commit
62fca3b
·
verified ·
1 Parent(s): f640e62

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -74
README.md CHANGED
@@ -1,81 +1,12 @@
1
  ---
2
- library_name: diffusers
3
- base_model: runwayml/stable-diffusion-v1-5
4
- model_type: text-to-image
5
  language:
6
  - en
7
- pipeline_tag: text-to-image
8
  tags:
 
 
 
9
  - art
10
- inference: true
11
- model:
12
- base_learning_rate: 1.0e-04
13
- target: ldm.models.diffusion.ddpm.LatentDiffusion
14
- params:
15
- linear_start: 0.00085
16
- linear_end: 0.0120
17
- num_timesteps_cond: 1
18
- log_every_t: 200
19
- timesteps: 1000
20
- first_stage_key: "image"
21
- cond_stage_key: "caption"
22
- image_size: 64
23
- channels: 4
24
- cond_stage_trainable: false # Note: different from the one we trained before
25
- conditioning_key: crossattn
26
- monitor: val/loss_simple_ema
27
- scale_factor: 0.18215
28
- use_ema: False
29
-
30
- scheduler_config: # 10000 warmup steps
31
- target: ldm.lr_scheduler.LambdaLinearScheduler
32
- params:
33
- warm_up_steps: [ 10000 ]
34
- cycle_lengths: [ 10000000000000 ] # incredibly large number to prevent corner cases
35
- f_start: [ 1.e-6 ]
36
- f_max: [ 1. ]
37
- f_min: [ 1. ]
38
-
39
- unet_config:
40
- target: ldm.modules.diffusionmodules.openaimodel.UNetModel
41
- params:
42
- image_size: 32 # unused
43
- in_channels: 4
44
- out_channels: 4
45
- model_channels: 320
46
- attention_resolutions: [ 4, 2, 1 ]
47
- num_res_blocks: 2
48
- channel_mult: [ 1, 2, 4, 4 ]
49
- num_heads: 8
50
- use_spatial_transformer: True
51
- transformer_depth: 1
52
- context_dim: 768
53
- use_checkpoint: True
54
- legacy: False
55
 
56
- first_stage_config:
57
- target: ldm.models.autoencoder.AutoencoderKL
58
- params:
59
- embed_dim: 4
60
- monitor: val/rec_loss
61
- ddconfig:
62
- double_z: true
63
- z_channels: 4
64
- resolution: 256
65
- in_channels: 3
66
- out_ch: 3
67
- ch: 128
68
- ch_mult:
69
- - 1
70
- - 2
71
- - 4
72
- - 4
73
- num_res_blocks: 2
74
- attn_resolutions: [ ]
75
- dropout: 0.0
76
- lossconfig:
77
- target: torch.nn.Identity
78
-
79
- cond_stage_config:
80
- target: ldm.modules.encoders.modules.FrozenCLIPEmbedder
81
  ---
 
1
  ---
 
 
 
2
  language:
3
  - en
 
4
  tags:
5
+ - stable-diffusion
6
+ - stable-diffusion-diffusers
7
+ - text-to-image
8
  - art
9
+ - diffusers
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
 
11
+ pipeline_tag: text-to-image
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  ---