Divyasreepat commited on
Commit
b547d81
·
verified ·
1 Parent(s): 0275605

Update README.md with new model card content

Browse files
Files changed (1) hide show
  1. README.md +2 -20
README.md CHANGED
@@ -1,24 +1,6 @@
1
  ---
2
  library_name: keras-hub
3
  ---
4
- This is a [`StableDiffusion3` model](https://keras.io/api/keras_hub/models/stable_diffusion3) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
5
- Model config:
6
- * **name:** stable_diffusion_3.5_backbone
7
- * **trainable:** True
8
- * **mmdit_patch_size:** 2
9
- * **mmdit_hidden_dim:** 2432
10
- * **mmdit_num_layers:** 38
11
- * **mmdit_num_heads:** 38
12
- * **mmdit_position_size:** 192
13
- * **mmdit_qk_norm:** rms_norm
14
- * **vae:** {'module': 'keras_hub.src.models.vae.vae_backbone', 'class_name': 'VAEBackbone', 'config': {'name': 'vae', 'trainable': True, 'encoder_num_filters': [128, 256, 512, 512], 'encoder_num_blocks': [2, 2, 2, 2], 'decoder_num_filters': [512, 512, 256, 128], 'decoder_num_blocks': [3, 3, 3, 3], 'sampler_method': 'sample', 'input_channels': 3, 'sample_channels': 32, 'output_channels': 3, 'scale': 1.5305, 'shift': 0.0609}, 'registered_name': 'VAEBackbone'}
15
- * **clip_l:** {'module': 'keras_hub.src.models.clip.clip_text_encoder', 'class_name': 'CLIPTextEncoder', 'config': {'name': 'clip_l', 'trainable': True, 'vocabulary_size': 49408, 'embedding_dim': 768, 'hidden_dim': 768, 'num_layers': 12, 'num_heads': 12, 'intermediate_dim': 3072, 'intermediate_activation': 'quick_gelu', 'intermediate_output_index': 10, 'max_sequence_length': 77}, 'registered_name': 'CLIPTextEncoder'}
16
- * **clip_g:** {'module': 'keras_hub.src.models.clip.clip_text_encoder', 'class_name': 'CLIPTextEncoder', 'config': {'name': 'clip_g', 'trainable': True, 'vocabulary_size': 49408, 'embedding_dim': 1280, 'hidden_dim': 1280, 'num_layers': 32, 'num_heads': 20, 'intermediate_dim': 5120, 'intermediate_activation': 'gelu', 'intermediate_output_index': 30, 'max_sequence_length': 77}, 'registered_name': 'CLIPTextEncoder'}
17
- * **t5:** None
18
- * **latent_channels:** 16
19
- * **output_channels:** 3
20
- * **num_train_timesteps:** 1000
21
- * **shift:** 3.0
22
- * **image_shape:** [1024, 1024, 3]
23
 
24
- This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
 
1
  ---
2
  library_name: keras-hub
3
  ---
4
+ ### Model Overview
5
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6