Shaoan commited on
Commit
322babc
·
verified ·
1 Parent(s): cc9a502

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 = './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(
 
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(