Update README.md

README.md

@@ -1,61 +1,71 @@
 ---
 widget:
-- text: "Jens Peter Hansen kommer fra Danmark"
-
+- text: Jens Peter Hansen kommer fra Danmark
 model:
   target: cldm.cldm.ControlLDM
   params:
     linear_start: 0.00085
-    linear_end: 0.0120
+    linear_end: 0.012
     num_timesteps_cond: 1
     log_every_t: 200
     timesteps: 1000
-    first_stage_key: "jpg"
-    cond_stage_key: "txt"
-    control_key: "hint"
+    first_stage_key: jpg
+    cond_stage_key: txt
+    control_key: hint
     image_size: 64
     channels: 4
     cond_stage_trainable: false
     conditioning_key: crossattn
     monitor: val/loss_simple_ema
     scale_factor: 0.18215
-    use_ema: False
-    only_mid_control: False
-
+    use_ema: false
+    only_mid_control: false
     control_stage_config:
       target: cldm.cldm.ControlNet
       params:
-        image_size: 32 # unused
+        image_size: 32
         in_channels: 4
         hint_channels: 3
         model_channels: 320
-        attention_resolutions: [ 4, 2, 1 ]
+        attention_resolutions:
+        - 4
+        - 2
+        - 1
         num_res_blocks: 2
-        channel_mult: [ 1, 2, 4, 4 ]
+        channel_mult:
+        - 1
+        - 2
+        - 4
+        - 4
         num_heads: 8
-        use_spatial_transformer: True
+        use_spatial_transformer: true
         transformer_depth: 1
         context_dim: 768
-        use_checkpoint: True
-        legacy: False
-
+        use_checkpoint: true
+        legacy: false
     unet_config:
       target: cldm.cldm.ControlledUnetModel
       params:
-        image_size: 32 # unused
+        image_size: 32
         in_channels: 4
         out_channels: 4
         model_channels: 320
-        attention_resolutions: [ 4, 2, 1 ]
+        attention_resolutions:
+        - 4
+        - 2
+        - 1
         num_res_blocks: 2
-        channel_mult: [ 1, 2, 4, 4 ]
+        channel_mult:
+        - 1
+        - 2
+        - 4
+        - 4
         num_heads: 8
-        use_spatial_transformer: True
+        use_spatial_transformer: true
         transformer_depth: 1
         context_dim: 768
-        use_checkpoint: True
-        legacy: False
-
+        use_checkpoint: true
+        legacy: false
     first_stage_config:
       target: ldm.models.autoencoder.AutoencoderKL
       params:
@@ -75,10 +85,10 @@ model:
           - 4
           num_res_blocks: 2
           attn_resolutions: []
-          dropout: 0.0
+          dropout: 0
         lossconfig:
           target: torch.nn.Identity
-
     cond_stage_config:
       target: ldm.modules.encoders.modules.FrozenCLIPEmbedder
+pipeline_tag: text-to-image
 ---