| Serialized with: |
|
|
| ```python |
| import torch |
| from transformers import CLIPTextConfig, CLIPTextModelWithProjection |
| |
| def get_dummy_components(): |
| clip_text_encoder_config = CLIPTextConfig( |
| bos_token_id=0, |
| eos_token_id=2, |
| hidden_size=32, |
| intermediate_size=37, |
| layer_norm_eps=1e-05, |
| num_attention_heads=4, |
| num_hidden_layers=5, |
| pad_token_id=1, |
| vocab_size=1000, |
| hidden_act="gelu", |
| projection_dim=32, |
| ) |
| |
| torch.manual_seed(0) |
| text_encoder = CLIPTextModelWithProjection(clip_text_encoder_config) |
| |
| torch.manual_seed(0) |
| text_encoder_2 = CLIPTextModelWithProjection(clip_text_encoder_config) |
| |
| return text_encoder, text_encoder_2 |
| |
| |
| text_encoder, text_encoder_2 = get_dummy_components() |
| text_encoder.push_to_hub("hf-internal-testing/tiny-sd3-text_encoder") |
| text_encoder_2.push_to_hub("hf-internal-testing/tiny-sd3-text_encoder-2") |
| ``` |
