Spaces:
Running
on
Zero
Running
on
Zero
Upload folder using huggingface_hub
Browse files- text_encoder.py +1 -1
text_encoder.py
CHANGED
|
@@ -393,7 +393,7 @@ class LoraT5Embedder(torch.nn.Module):
|
|
| 393 |
self.use_gradient_checkpointing = use_gradient_checkpointing
|
| 394 |
dtype = torch.bfloat16
|
| 395 |
self.dtype = dtype
|
| 396 |
-
t5_version = '
|
| 397 |
self.t5_tokenizer = T5Tokenizer.from_pretrained(t5_version, max_length=max_length)
|
| 398 |
|
| 399 |
self.t5_encoder = T5EncoderModel.from_pretrained(
|
|
|
|
| 393 |
self.use_gradient_checkpointing = use_gradient_checkpointing
|
| 394 |
dtype = torch.bfloat16
|
| 395 |
self.dtype = dtype
|
| 396 |
+
t5_version = 'google/t5-v1_1-xxl'
|
| 397 |
self.t5_tokenizer = T5Tokenizer.from_pretrained(t5_version, max_length=max_length)
|
| 398 |
|
| 399 |
self.t5_encoder = T5EncoderModel.from_pretrained(
|