sayakpaul's picture
sayakpaul HF staff
Create README.md
12e47ac verified
|
raw
history blame contribute delete
No virus
932 Bytes
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")
```