Update README.md
Browse files
README.md
CHANGED
|
@@ -71,7 +71,7 @@ def get_dummy_components(num_layers: int = 1, num_single_layers: int = 1):
|
|
| 71 |
text_encoder_2 = T5EncoderModel(config=config)
|
| 72 |
|
| 73 |
tokenizer = CLIPTokenizer.from_pretrained("hf-internal-testing/tiny-random-clip")
|
| 74 |
-
tokenizer_2 = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
|
| 75 |
|
| 76 |
torch.manual_seed(0)
|
| 77 |
vae = AutoencoderKL(
|
|
|
|
| 71 |
text_encoder_2 = T5EncoderModel(config=config)
|
| 72 |
|
| 73 |
tokenizer = CLIPTokenizer.from_pretrained("hf-internal-testing/tiny-random-clip")
|
| 74 |
+
tokenizer_2 = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5", model_max_length=512)
|
| 75 |
|
| 76 |
torch.manual_seed(0)
|
| 77 |
vae = AutoencoderKL(
|